generate tests updated for toytree

dereneaton · Jan 7, 2025 · 9e7dd5a · 9e7dd5a
1 parent 6b6c329
commit 9e7dd5a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ipyrad/analysis/baba2.py b/ipyrad/analysis/baba2.py
@@ -857,10 +857,10 @@ def __init__(self, tree, constraint_dict, constraint_exact):
 
         # tree to traverse
         if isinstance(tree, toytree.ToyTree):
-            self.tree = tree.copy()
+            tree = tree.copy()
         else:
-            self.tree = toytree.tree(tree)
-        if not self.tree.is_rooted(): 
+            tree = toytree.tree(tree)
+        if not tree.is_rooted(): 
             raise IPyradError(
                 "generate_tests_from_tree(): tree must be rooted and resolved")
 
@@ -895,7 +895,7 @@ def __init__(self, tree, constraint_dict, constraint_exact):
         self.xdict = np.array(self.xdict).astype(bool)
 
         # get tests
-        self.loop(self.tree.treenode)
+        self.loop(tree.treenode)
 
         # order and check redundancy
         tests = []