diff --git a/python/tests/beagle_numba.py b/python/tests/beagle_numba.py index 8ddecffc9f..bee828783f 100644 --- a/python/tests/beagle_numba.py +++ b/python/tests/beagle_numba.py @@ -588,7 +588,7 @@ def interpolate_allele_probs( probs[i, a] += (1 - w) * state_mat[k, j] # Rescale probabilities. # TODO: Check if this is necessary. Could this be a subtle source of error? - assert np.all(np.sum(probs, axis=1) > 0), "Sum of allele probabilities is not >= 0." + assert np.all(np.sum(probs, axis=1) > 0), "Sum of allele probabilities is not > 0." probs_rescaled = probs / np.sum(probs, axis=1)[:, np.newaxis] if return_weights: return (probs_rescaled, weights)