Fix bug

Author: szhan

python/tests/beagle.py

@@ -202,10 +202,9 @@ def compute_forward_probability_matrix(ref_h, query_h, rho, mu):
             ref_a = ref_h[i, j]
             # Get emission probability.
             em_prob = mu[i] if query_a != ref_a else 1.0 - mu[i]
-            if i == 0:
-                fm[i, j] = em_prob
-            else:
-                fm[i, j] = em_prob * (scale * fm[i - 1, j] + shift)
+            fm[i, j] = em_prob
+            if i > 0:
+                fm[i, j] *= scale * fm[i - 1, j] + shift
         site_sum = np.sum(fm[i, :])
         last_sum = site_sum / h if i == 0 else site_sum
     return fm

python/tests/beagle_numba.py

@@ -144,12 +144,11 @@ def compute_forward_probability_matrix(ref_h, query_h, rho, mu):
             ref_a = ref_h[i, j]
             # Get emission probability
             em_prob = mu[i] if query_a != ref_a else 1.0 - mu[i]
-            if m == 0:
-                fm[i, j] = em_prob
-            else:
-                fm[i, j] = em_prob * (scale * fm[i - 1, j] + shift)
+            fm[i, j] = em_prob
+            if i > 0:
+                fm[i, j] *= scale * fm[i - 1, j] + shift
         site_sum = np.sum(fm[i, :])
-        last_sum = site_sum / h if m == 0 else site_sum
+        last_sum = site_sum / h if i == 0 else site_sum
     return fm