MAINT: Directly use np.ndarray.__class_getitem__ for the definition…

… of `npt.NDArray`
numpy · Sep 15, 2021 · 0baeeb1 · 0baeeb1
1 parent 49f5358
commit 0baeeb1
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/numpy/typing/_generic_alias.py b/numpy/typing/_generic_alias.py
@@ -205,12 +205,9 @@ def __getattribute__(self, name: str) -> Any:
 
 ScalarType = TypeVar("ScalarType", bound=np.generic, covariant=True)
 
-if TYPE_CHECKING:
+if TYPE_CHECKING or sys.version_info >= (3, 9):
     _DType = np.dtype[ScalarType]
     NDArray = np.ndarray[Any, np.dtype[ScalarType]]
-elif sys.version_info >= (3, 9):
-    _DType = types.GenericAlias(np.dtype, (ScalarType,))
-    NDArray = types.GenericAlias(np.ndarray, (Any, _DType))
 else:
     _DType = _GenericAlias(np.dtype, (ScalarType,))
     NDArray = _GenericAlias(np.ndarray, (Any, _DType))