Rename variables

szhan · Feb 23, 2024 · fe5f645 · fe5f645
1 parent be3742e
commit fe5f645
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/tests/beagle_numba.py b/python/tests/beagle_numba.py
@@ -539,8 +539,8 @@ def run_tsimpute(
     typed_cm = convert_to_genetic_map_positions(typed_pos, genetic_map)
     untyped_cm = convert_to_genetic_map_positions(untyped_pos, genetic_map)
     # Get rates at genotyped markers.
-    rho = rho[typed_idx]
-    mu = mu[typed_idx]
+    transition_probs = rho[typed_idx]
+    mismatch_probs = mu[typed_idx]
     # Subset haplotypes.
     ref_ts_typed = ref_ts.delete_sites(site_ids=untyped_idx)
     ref_ts_untyped = ref_ts.delete_sites(site_ids=typed_idx)
@@ -552,8 +552,8 @@ def run_tsimpute(
     # WARN: Be careful with argument `acgt_alleles`!!!
     ls_hmm = _tskit.LsHmm(
         ref_ts_typed._ll_tree_sequence,
-        recombination_rate=rho,
-        mutation_rate=mu,
+        recombination_rate=transition_probs,  # Transition probabilities
+        mutation_rate=mismatch_probs,  # Mismatch probabilities
         acgt_alleles=True,
         precision=precision,
     )