Fix TestRealTimeKernel::test_explicit_shuffle (#1252)

theislab · Feb 13, 2025 · 9c393e7 · 9c393e7
1 parent f61590a
commit 9c393e7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_kernels.py b/tests/test_kernels.py
@@ -1209,7 +1209,8 @@ def test_explicit_shuffle(self, adata_large: AnnData):
         tmat = tmk.transition_matrix
 
         for src, tgt in zip(cats[:-1], cats[1:]):
-            src_mask, tgt_mask = tmk.time == src, tmk.time == tgt
+            src_mask = (tmk.time == src).tolist()
+            tgt_mask = (tmk.time == tgt).tolist()
             np.testing.assert_allclose(tmat[src_mask, :][:, tgt_mask].toarray(), expected[src, tgt])
 
     @pytest.mark.parametrize(