Correct compute_forward_matrix

szhan · Mar 4, 2024 · 0581375 · 0581375
1 parent fe70e44
commit 0581375
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/tests/beagle_numba.py b/python/tests/beagle_numba.py
@@ -357,7 +357,8 @@ def compute_forward_matrix(ref_h, query_h, trans_probs, mismatch_probs):
             if query_a == ref_a:
                 em_prob = 1.0 - mismatch_probs[i]
             fwd_mat[i, j] = em_prob
-            fwd_mat[i, j] *= scale * fwd_mat[i - 1, j] + shift
+            if i > 0:
+                fwd_mat[i, j] *= scale * fwd_mat[i - 1, j] + shift
         site_sum = np.sum(fwd_mat[i, :])
         # Prior probabilities are multiplied when i = 0.
         last_sum = site_sum / h if i == 0 else site_sum