diff --git a/array_api_strict/_array_object.py b/array_api_strict/_array_object.py index b356b82..a3a386e 100644 --- a/array_api_strict/_array_object.py +++ b/array_api_strict/_array_object.py @@ -429,7 +429,7 @@ def _validate_index( isinstance(i, SupportsIndex) # i.e. ints or isinstance(i, slice) or i == Ellipsis - or i is None + or (op == "getitem" and i is None) # `None` disallowed in setitem or isinstance(i, Array) or isinstance(i, np.ndarray) ):