diff --git a/array_api_strict/_manipulation_functions.py b/array_api_strict/_manipulation_functions.py index fe4a608..7c4adda 100644 --- a/array_api_strict/_manipulation_functions.py +++ b/array_api_strict/_manipulation_functions.py @@ -138,6 +138,10 @@ def roll( See its docstring for more information. """ + if not isinstance(shift, int | tuple): + raise ValueError( + f"`shift` can only be an int or a tuple, got {type(shift)=} instead." + ) return Array._new(np.roll(x._array, shift, axis=axis), device=x.device)