Use tskit constant

szhan · Feb 19, 2024 · f53c008 · f53c008
1 parent 4906434
commit f53c008
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tests/beagle_numba.py b/python/tests/beagle_numba.py
@@ -369,7 +369,7 @@ def interpolate_allele_probs(sm, ref_h, typed_pos, untyped_pos, typed_cm, untype
     :rtype: numpy.ndarray
     """
     # TODO: Allow for biallelic site matrix. Work with `_tskit.lshmm` properly.
-    alleles = np.arange(4)  # ACGT
+    alleles = np.arange(len(tskit.ALLELES_ACGT))
     m = sm.shape[0]
     x = ref_h.shape[0]
     h = ref_h.shape[1]