Fix dtype.

vnmabus · Jan 26, 2024 · 34cb6e0 · 34cb6e0
1 parent 9202110
commit 34cb6e0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dcor/tests/test_dcor.py b/dcor/tests/test_dcor.py
@@ -517,8 +517,8 @@ def test_integer_overflow(self) -> None:
         n_samples = 10000
 
         # some simple data
-        arr1 = np.array([1, 2, 3] * n_samples)
-        arr2 = np.array([10, 20, 5] * n_samples)
+        arr1 = np.array([1, 2, 3] * n_samples, dtype=np.int32)
+        arr2 = np.array([10, 20, 5] * n_samples, dtype=np.int32)
 
         int_int = dcor.distance_correlation(
             arr1,