Fancy indexing in __getitem__ signature

Author: crusaderky

array_api_strict/_array_object.py

@@ -351,8 +351,8 @@ def _validate_index(
             int
             | slice
             | EllipsisType
-            | tuple[int | slice | EllipsisType | None, ...]
             | Array
+            | tuple[int | slice | EllipsisType | Array | None, ...]
         ),
         op: Literal["getitem", "setitem"] = "getitem",
     ) -> None:
@@ -711,8 +711,9 @@ def __getitem__(
             int
             | slice
             | EllipsisType
-            | tuple[int | slice | EllipsisType | None, ...]
             | Array
+            | None
+            | tuple[int | slice | EllipsisType | Array | None, ...]
         ),
         /,
     ) -> Array:
@@ -926,8 +927,14 @@ def __rshift__(self, other: Array | int, /) -> Array:
 
     def __setitem__(
         self,
-        key: (  # Almost same as __getitem__ key but doesn't accept None
-            int | slice | EllipsisType | tuple[int | slice | EllipsisType, ...] | Array
+        # Almost same as __getitem__ key but doesn't accept None
+        # or integer arrays
+        key: (
+            int
+            | slice
+            | EllipsisType
+            | Array
+            | tuple[int | slice | EllipsisType, ...]
         ),
         value: Array | complex,
         /,