Add protocol for namespace

Signed-off-by: nstarman <[email protected]>

Author: nstarman

Diff for: src/array_api_stubs/_draft/_namespace.py

@@ -0,0 +1,8 @@
+__all__ = ["ArrayAPINamespace"]
+
+from typing import Protocol
+
+
+class ArrayAPINamespace(Protocol):
+    """Protocol for the array API namespace itself."""
+

Diff for: src/array_api_stubs/_draft/array_object.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+from typing import TYPE_CHECKING
+
 from ._types import (
     array,
     dtype as Dtype,
@@ -13,6 +15,9 @@
     ellipsis,
 )
 
+if TYPE_CHECKING:
+    from ._namespace import ArrayAPINamespace
+
 
 class _array:
     def __init__(self: array) -> None:
@@ -195,7 +200,7 @@ def __and__(self: array, other: Union[int, bool, array], /) -> array:
 
     def __array_namespace__(
         self: array, /, *, api_version: Optional[str] = None
-    ) -> Any:
+    ) -> ArrayAPINamespace:
         """
         Returns an object that has all the array API functions on it.