Fix bug

szhan · Sep 13, 2023 · 2f0f3ce · 2f0f3ce
1 parent 3830c64
commit 2f0f3ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tests/beagle.py b/python/tests/beagle.py
@@ -516,7 +516,7 @@ def interpolate_allele_probabilities(sm, ref_h, genotyped_pos, imputed_pos):
     rescaled_p = np.zeros_like(p)
     for i in np.arange(x):
         rescaled_p[i, :] = p[i, :] / np.sum(p[i, :])
-    return p
+    return rescaled_p
 
 
 def get_map_alleles(allele_probs):