Add njit

szhan · Mar 4, 2024 · 8f26dc5 · 8f26dc5
1 parent 96f944a
commit 8f26dc5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/tests/beagle_numba.py b/python/tests/beagle_numba.py
@@ -322,6 +322,7 @@ def get_transition_probs(cm, h, ne):
     return trans_probs
 
 
+@njit
 def compute_emission_probability(mismatch_prob, is_match, *, num_alleles=2):
     """
     Compute emission probability at a site based on whether the alleles