From 433f12d797ca344332fe6e9e4aa5333760ea75f5 Mon Sep 17 00:00:00 2001
From: Shing Zhan <shing.zhan@gmail.com>
Date: Tue, 30 Jan 2024 04:42:00 -0800
Subject: [PATCH] Minor refactor

---
 python/tests/beagle_numba.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/python/tests/beagle_numba.py b/python/tests/beagle_numba.py
index c6d54f2505..63dcea8d69 100644
--- a/python/tests/beagle_numba.py
+++ b/python/tests/beagle_numba.py
@@ -100,11 +100,10 @@ def check_data(ref_h, query_h):
     :return: True if alleles in query are in reference.
     :rtype: bool
     """
-    _MISSING = -1
     m = ref_h.shape[0]
     assert m == len(query_h), "Reference and query differ in length."
     for i in range(m):
-        if query_h[i] == _MISSING:
+        if query_h[i] == tskit.MISSING_DATA:
             continue
         else:
             if query_h[i] not in ref_h[i, :]: