force nonnegative

KoslickiLab · Oct 2, 2023 · 1140319 · 1140319
1 parent 2735872
commit 1140319
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/fununifrac/reproducibility/Kegg_tree.py b/fununifrac/reproducibility/Kegg_tree.py
@@ -273,9 +273,13 @@ def solve_branch_lengths(self, edge_length_solutions, level):
                     d2 = self.needed_pairs[level][(node, another)]
                     d3 = self.needed_pairs[level][(sib, another)]
                     e1 = (d1+d2-d3)/2
+                    if e1 <= 0:
+                        e1 = sys.float_info.epsilon
                     edge_length_solutions[(parent, node)] = e1
                     if (parent, sib) not in edge_length_solutions:
                         e2 = d1 - e1
+                        if e2 <= 0:
+                            e2 = sys.float_info.epsilon
                         edge_length_solutions[(parent, sib)] = e2
         self.update_needed_pairs(level-1, edge_length_solutions)
         self.solve_branch_lengths(edge_length_solutions, level-1)