Merge pull request #33 from nossleinad/getnodelist/return-type

Type getnodelist and co. return arrays
MurrellGroup · Aug 5, 2024 · cb66020 · cb66020
2 parents 44a060d + 64efb0a
commit cb66020
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/core/nodes/AbstractTreeNode.jl b/src/core/nodes/AbstractTreeNode.jl
@@ -348,7 +348,7 @@ end
 
 export getnodelist
 function getnodelist(node::T) where {T<:AbstractTreeNode}
-    nodelist = []
+    nodelist = T[]
     nodes = [node]
     while nodes != []
         node = pop!(nodes)
@@ -391,7 +391,7 @@ end
 
 export getnonleaflist
 function getnonleaflist(node::T) where {T<:AbstractTreeNode}
-    nonleaflist = []
+    nonleaflist = T[]
     nodes = [node]
     while nodes != []
         node = pop!(nodes)
@@ -407,7 +407,7 @@ end
 
 export getleaflist
 function getleaflist(node::T) where {T<:AbstractTreeNode}
-    leaflist = []
+    leaflist = T[]
     nodes = [node]
     while nodes != []
         node = pop!(nodes)