TYP: accept review comment on ignoring NotImplemented in type checking

Co-authored-by: Bas van Beek <[email protected]>

Original NumPy Commit: 890c0cfe15f0e3325b57e44f9b9a9f78f9797f25

Author: rgommers

array_api_strict/_array_object.py

@@ -125,10 +125,7 @@ def __array__(self, dtype: None | np.dtype[Any] = None) -> npt.NDArray[Any]:
     # spec in places where it either deviates from or is more strict than
     # NumPy behavior
 
-    # NOTE: no valid type annotation possible.  E.g `Union[Array,
-    # NotImplemented]` is forbidden, see https://github.com/python/mypy/issues/363
-    # Maybe change returned object to `Literal['NotImplemented']`?
-    def _check_allowed_dtypes(self, other, dtype_category, op):
+    def _check_allowed_dtypes(self, other: bool | int | float | Array, dtype_category: str, op: str) -> Array:
         """
         Helper function for operators to only allow specific input dtypes