Vectorise

tskit-dev · Sep 1, 2023 · 3cbff90 · 3cbff90
1 parent 7195b73
commit 3cbff90
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/tests/beagle.py b/python/tests/beagle.py
@@ -241,8 +241,7 @@ def compute_backward_probability_matrix(ref_h, query_h, rho, mu):
         sum_site = np.sum(bm[i, :])
         scale = (1 - rho[i + 1]) / sum_site
         shift = rho[i + 1] / h
-        for j in np.arange(h):
-            bm[i, j] = scale * bm[i, j] + shift
+        bm[i, :] = scale * bm[i, :] + shift
     return bm