Skip to content

Commit 06e64ff

Browse files
committed
Fix dtype check for ILP64
1 parent 4e20dc7 commit 06e64ff

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

sparse_dot_mkl/tests/test_service.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,9 @@ def test_set_interface_layer(self):
4545
mkl_set_interface_layer("MKL")
4646

4747
def test_get_integer_interface(self):
48-
self.assertEqual(
49-
np.int32,
50-
mkl_interface_integer_dtype()
48+
self.assertTrue(
49+
mkl_interface_integer_dtype() in [
50+
np.int32,
51+
np.int64
52+
]
5153
)

0 commit comments

Comments
 (0)