Fix run_beagle

tskit-dev · Aug 26, 2023 · 5d9a9c1 · 5d9a9c1
1 parent f626e70
commit 5d9a9c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/tests/test_beagle.py b/python/tests/test_beagle.py
@@ -365,8 +365,8 @@ def run_beagle(ref_h, query_h, genotyped_pos, imputed_pos):
     :return: MAP alleles at imputed markers in the query haplotype.
     :rtype: numpy.ndarray
     """
-    mu = get_mismatch_prob()
-    rho = get_switch_prob()
+    mu = get_mismatch_prob(genotyped_pos)
+    rho = get_switch_prob(genotyped_pos)
     fm = compute_forward_probability_matrix(ref_h, query_h, rho, mu)
     bm = compute_backward_probability_matrix(ref_h, query_h, rho, mu)
     _, fwd_hap_probs, bwd_hap_probs = compute_state_probability_matrix(