From 35e34643d4dcfa346e8ad398b0e3843601a60590 Mon Sep 17 00:00:00 2001 From: Amanda Potts Date: Tue, 10 Sep 2024 12:51:43 -0400 Subject: [PATCH] fix failing tests --- tests/array_api/array_creation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/array_api/array_creation.py b/tests/array_api/array_creation.py index 2167d7326d..f12f9094d1 100644 --- 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: