Improve doc

tskit-dev · Aug 31, 2023 · 62e8ed8 · 62e8ed8
1 parent 47b8ccd
commit 62e8ed8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/tests/test_beagle.py b/python/tests/test_beagle.py
@@ -110,8 +110,9 @@ def get_mismatch_prob(pos, miscall_rate=0.0001):
     :return: Mismatch probabilities.
     :rtype: numpy.ndarray
     """
+    assert isinstance(miscall_rate, float)
     assert 0 <= miscall_rate <= 0.5
-    mu = np.repeat(miscall_rate, len(pos))
+    mu = np.zeros(len(pos), dtype=np.float64) + miscall_rate
     return mu