fix: outlier pruning in clock-filter

neherlab · Jul 12, 2024 · 800deec · 800deec
1 parent 338f379
commit 800deec
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/treetime/wrappers.py b/treetime/wrappers.py
@@ -859,7 +859,8 @@ def estimate_clock_model(params):
         myTree.outliers.to_csv(basename + 'outliers.tsv', sep='\t')
 
     if hasattr(myTree, 'outliers') and myTree.outliers is not None and params.prune_outliers:
-        for outlier in myTree.outliers:
+        for outlier in myTree.outliers.index:
+            print("removing ", outlier)
             myTree.tree.prune(outlier)
 
     if not params.keep_root: