Remove unnecessary astype

szhan · Sep 20, 2023 · 89619b8 · 89619b8
1 parent 2bc217d
commit 89619b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tests/beagle_numba.py b/python/tests/beagle_numba.py
@@ -342,7 +342,7 @@ def run_tsimpute(ref_ts, query_h, pos, mu, rho):
     ref_ts_genotyped = ref_ts.delete_sites(site_ids=ungenotyped_site_idx)
     ref_ts_ungenotyped = ref_ts.delete_sites(site_ids=genotyped_site_idx)
     ref_h_ungenotyped = ref_ts_ungenotyped.genotype_matrix()
-    query_h_genotyped = query_h[genotyped_site_idx].astype(np.int32)
+    query_h_genotyped = query_h[genotyped_site_idx]
     # Get forward and backward matrices from tree sequence.
     fm = _tskit.CompressedMatrix(ref_ts_genotyped._ll_tree_sequence)
     bm = _tskit.CompressedMatrix(ref_ts_genotyped._ll_tree_sequence)