fix failing tests

Bears-R-Us · Sep 10, 2024 · 35e3464 · 35e3464
1 parent ac22c11
commit 35e3464
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/array_api/array_creation.py b/tests/array_api/array_creation.py
@@ -59,7 +59,7 @@ def test_triu(self, data_type, prob_size):
         # test on one square and two non-square matrices
 
         for rows, cols in [(size, size), (size + 1, size - 1), (size - 1, size + 1)]:
-            pda = ak.randint(1, 10, (rows, cols))
+            pda = xp.asarray(ak.randint(1, 10, (rows, cols)))
             nda = pda.to_ndarray()
             sweep = range(-(rows - 1), cols - 1)  # sweeps the diagonal from LL to UR
             for diag in sweep:
@@ -78,7 +78,7 @@ def test_tril(self, data_type, prob_size):
         # test on one square and two non-square matrices
 
         for rows, cols in [(size, size), (size + 1, size - 1), (size - 1, size + 1)]:
-            pda = ak.randint(1, 10, (rows, cols))
+            pda = xp.asarray(ak.randint(1, 10, (rows, cols)))
             nda = pda.to_ndarray()
             sweep = range(-(rows - 2), cols)  # sweeps the diagonal from LL to UR
             for diag in sweep: