Move all typing imports under "if TYPE_CHECKING"

Author: asmeurer

array_api_strict/_array_object.py

@@ -33,11 +33,12 @@
 )
 from ._flags import get_array_api_strict_flags, set_array_api_strict_flags
 
-from typing import TYPE_CHECKING, Optional, Tuple, Union, Any, SupportsIndex
+from typing import TYPE_CHECKING, SupportsIndex
 import types
 
 if TYPE_CHECKING:
-    from ._typing import Any, PyCapsule, Device, Dtype
+    from typing import Optional, Tuple, Union, Any
+    from ._typing import PyCapsule, Device, Dtype
     import numpy.typing as npt
 
 import numpy as np

array_api_strict/_data_type_functions.py

@@ -15,9 +15,10 @@
 )
 
 from dataclasses import dataclass
-from typing import TYPE_CHECKING, List, Tuple, Union
+from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
+    from typing import List, Tuple, Union
     from ._typing import Dtype
 
 import numpy as np

array_api_strict/_indexing_functions.py

@@ -3,7 +3,10 @@
 from ._array_object import Array
 from ._dtypes import _integer_dtypes
 
-from typing import Optional
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from typing import Optional
 
 import numpy as np
 

array_api_strict/_manipulation_functions.py

@@ -3,7 +3,10 @@
 from ._array_object import Array
 from ._data_type_functions import result_type
 
-from typing import List, Optional, Tuple, Union
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from typing import List, Optional, Tuple, Union
 
 import numpy as np
 

array_api_strict/_searching_functions.py

@@ -4,7 +4,9 @@
 from ._dtypes import _result_type, _real_numeric_dtypes
 from ._flags import requires_data_dependent_shapes
 
-from typing import Optional, Tuple
+from typing import TYPE_CHECKING
+if TYPE_CHECKING:
+    from typing import Optional, Tuple
 
 import numpy as np
 

array_api_strict/_statistical_functions.py

@@ -8,9 +8,10 @@
 from ._array_object import Array
 from ._dtypes import float32, complex64
 
-from typing import TYPE_CHECKING, Optional, Tuple, Union
+from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
+    from typing import Optional, Tuple, Union
     from ._typing import Dtype
 
 import numpy as np

array_api_strict/_utility_functions.py

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

array_api_strict/fft.py

@@ -1,8 +1,9 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Union, Optional, Literal
+from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
+    from typing import Union, Optional, Literal
     from ._typing import Device
     from collections.abc import Sequence