Use tskit constant for missing data

szhan · Feb 19, 2024 · 0192155 · 0192155
1 parent f53c008
commit 0192155
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/tests/beagle_numba.py b/python/tests/beagle_numba.py
@@ -457,8 +457,8 @@ def run_interpolation_beagle(
         stacklevel=1,
     )
     # Set marker indices.
-    typed_idx = np.where(query_h != -1)[0]
-    untyped_idx = np.where(query_h == -1)[0]
+    typed_idx = np.where(query_h != tskit.MISSING_DATA)[0]
+    untyped_idx = np.where(query_h == tskit.MISSING_DATA)[0]
     # Set physical positions of markers.
     typed_pos = pos[typed_idx]
     untyped_pos = pos[untyped_idx]
@@ -530,8 +530,8 @@ def run_tsimpute(
         stacklevel=1,
     )
     # Set markers indices.
-    typed_idx = np.where(query_h != -1)[0]
-    untyped_idx = np.where(query_h == -1)[0]
+    typed_idx = np.where(query_h != tskit.MISSING_DATA)[0]
+    untyped_idx = np.where(query_h == tskit.MISSING_DATA)[0]
     # Set physical positions of markers.
     typed_pos = pos[typed_idx]
     untyped_pos = pos[untyped_idx]