TYP: change type annotation for __array_namespace__ to ModuleType

This is more precise, we are returning a module here. Type checkers
will be able to use this info in the future - see
data-apis/array-api#267

Author: rgommers

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