Merge pull request #20725 from charris/backport-20719

charris · web-flow · commit d39ad3a68380 · 2022-01-04T10:54:58.000-07:00
TYP: change type annotation for `__array_namespace__` to ModuleType
diff --git a/numpy/array_api/_array_object.py b/numpy/array_api/_array_object.py
@@ -30,6 +30,7 @@
 )
 
 from typing import TYPE_CHECKING, Optional, Tuple, Union, Any
+import types
 
 if TYPE_CHECKING:
     from ._typing import Any, PyCapsule, Device, Dtype
@@ -415,7 +416,7 @@ def __and__(self: Array, other: Union[int, bool, Array], /) -> Array:
 
     def __array_namespace__(
         self: Array, /, *, api_version: Optional[str] = None
-    ) -> Any:
+    ) -> types.ModuleType:
         if api_version is not None and not api_version.startswith("2021."):
             raise ValueError(f"Unrecognized array API version: {api_version!r}")
         return array_api
