Hacky bypass to avoid vanishing scale

tskit-dev · Dec 5, 2024 · 1791b09 · 1791b09
1 parent b0e1126
commit 1791b09
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tsdate/variational.py b/tsdate/variational.py
@@ -443,11 +443,12 @@ def twin_projection(x, y):
                 factors[:, ROOTWARD] *= scale[edges_parent, np.newaxis]
                 factors[:, LEAFWARD] *= scale[edges_child, np.newaxis]
                 scale[:] = 1.0
+            # /FIXME
         # FIXME: this is a hacky bypass
-        if scale[p] < TINY or scale[c] < TINY:  # DEBUG
-            factors[:, ROOTWARD] *= scale[edges_parent, np.newaxis]
-            factors[:, LEAFWARD] *= scale[edges_child, np.newaxis]
-            scale[:] = 1.0
+        factors[:, ROOTWARD] *= scale[edges_parent, np.newaxis]
+        factors[:, LEAFWARD] *= scale[edges_child, np.newaxis]
+        scale[:] = 1.0
+        # /FIXME
 
     @staticmethod
     @numba.njit(_void(_b1r, _f2w, _f3w, _f1w, _f, _i, _f))