diff options
Diffstat (limited to 'numpy/typing/_generic_alias.py')
| -rw-r--r-- | numpy/typing/_generic_alias.py | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/numpy/typing/_generic_alias.py b/numpy/typing/_generic_alias.py index 5ad5e580c..932f12dd0 100644 --- a/numpy/typing/_generic_alias.py +++ b/numpy/typing/_generic_alias.py @@ -205,12 +205,9 @@ else: 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)) |