Fixed numpy 2.4.2 ValueError: setting an array element with a sequence.

Author: seanlaw

tests/test_core.py

@@ -1208,7 +1208,11 @@ def test_merge_topk_PI_without_overlap():
         PB = np.random.rand(n * k).reshape(n, k)
         col_idx = np.random.randint(0, k, size=n)
         for i in range(n):  # creating ties between values of PA and PB
-            PB[i, col_idx[i]] = np.random.choice(PA[i], size=1, replace=False)
+            val = np.random.choice(PA[i], size=1, replace=False)
+            try:
+                PB[i, col_idx[i]] = val
+            except ValueError:  # pragma: no cover
+                PB[i, col_idx[i]] = val.item()
         PB[:, :] = np.sort(PB, axis=1)  # sorting each row separately
 
         IA = np.arange(n * k).reshape(n, k)
@@ -1332,7 +1336,11 @@ def test_merge_topk_ρI_without_overlap():
         ρB = np.random.rand(n * k).reshape(n, k)
         col_idx = np.random.randint(0, k, size=n)
         for i in range(n):  # creating ties between values of PA and PB
-            ρB[i, col_idx[i]] = np.random.choice(ρA[i], size=1, replace=False)
+            val = np.random.choice(ρA[i], size=1, replace=False)
+            try:
+                ρB[i, col_idx[i]] = val
+            except ValueError:  # pragma: no cover
+                ρB[i, col_idx[i]] = val.item()
         ρB[:, :] = np.sort(ρB, axis=1)  # sorting each row separately
 
         IA = np.arange(n * k).reshape(n, k)