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
@@ -32,6 +32,7 @@ from ._dtypes import (
)
from typing import TYPE_CHECKING, Optional, Tuple, Union, Any
+import types
if TYPE_CHECKING:
from ._typing import Any, PyCapsule, Device, Dtype
@@ -419,7 +420,7 @@ class 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
