Remove typing exports from numpy/_array_api/_types.py

Author: asmeurer

numpy/_array_api/_creation_functions.py

@@ -1,11 +1,10 @@
 from __future__ import annotations
 
 
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, List, Optional, Tuple, Union
 if TYPE_CHECKING:
-    from ._types import (List, Optional, SupportsDLPack,
-                         SupportsBufferProtocol, Tuple, Union, Array, Device,
-                         Dtype)
+    from ._types import (NestedSequence, SupportsDLPack,
+                         SupportsBufferProtocol, Array, Device, Dtype)
     from collections.abc import Sequence
 from ._dtypes import _all_dtypes
 

numpy/_array_api/_manipulation_functions.py

@@ -2,9 +2,7 @@
 
 from ._array_object import Array
 
-from typing import TYPE_CHECKING
-if TYPE_CHECKING:
-    from ._types import Optional, Tuple, Union
+from typing import List, Optional, Tuple, Union
 
 import numpy as np
 

numpy/_array_api/_types.py

@@ -6,11 +6,10 @@
 valid for inputs that match the given type annotations.
 """
 
-__all__ = ['Any', 'List', 'Literal', 'Optional', 'Tuple', 'Union', 'Array',
-           'Device', 'Dtype', 'SupportsDLPack', 'SupportsBufferProtocol',
-           'PyCapsule']
+__all__ = ['Array', 'Device', 'Dtype', 'SupportsDLPack',
+           'SupportsBufferProtocol', 'PyCapsule']
 
-from typing import Any, List, Literal, Optional, Tuple, Union, TypeVar
+from typing import Any, Sequence, Type, Union
 
 from . import (Array, int8, int16, int32, int64, uint8, uint16, uint32,
                uint64, float32, float64)