Comments

Author: NeilGirdhar

src/array_api_typing/_array.py

@@ -1,8 +1,9 @@
 """Static typing support for the array API standard."""
 
-from typing import Protocol, final
+from typing import Protocol
+
+from ._namespace import HasArrayNamespace
 
 
-@final
 class Array(HasArrayNamespace, Protocol):
     pass

src/array_api_typing/_namespace.py

@@ -1,6 +1,6 @@
 """Static typing support for the array API standard."""
 
-from typing import Protocol, final
+from typing import Protocol
 from typing_extensions import TypeVar
 
 from ._array import Array
@@ -41,10 +41,9 @@ def astype(
     ) -> A: ...
 
 
-T = TypeVar("T", bound=Namespace, default=Namespace)  # PEP 696 default
+T = TypeVar("T", bound=ArrayNamespace, default=ArrayNamespace)  # PEP 696 default
 
 
-@final
 class HasArrayNamespace(Protocol[T]):  # type: ignore[misc]  # see python/mypy#17288
     """Protocol for classes that have an `__array_namespace__` method.
 

src/array_api_typing/_simple.py

@@ -1,5 +1,7 @@
 """Static typing support for the array API standard."""
 
+from __future__ import annotations
+
 from typing import Any, Protocol, TypeAlias, TypeVar
 
 _T_co = TypeVar("_T_co", covariant=True)