diff --git a/array_api_tests/test_manipulation_functions.py b/array_api_tests/test_manipulation_functions.py index 754b507d..13412a30 100644 --- a/array_api_tests/test_manipulation_functions.py +++ b/array_api_tests/test_manipulation_functions.py @@ -341,7 +341,7 @@ def test_repeat(x, kw, data): for i, count in enumerate(repeats_array): end = start + count ph.assert_array_elements("repeat", out=out_slice[start:end], - expected=xp.full((count,), x_slice[i], dtype=x.dtype), + expected=xp.full((int(count),), x_slice[i], dtype=x.dtype), kw=kw) start = end