Merge pull request numpy#27292 from DimitriPapadopoulos/TCH

MAINT: Start applying ruff/flake8-type-checking rules (TCH)

Author: charris

numpy/_typing/_nested_sequence.py

@@ -2,14 +2,17 @@
 
 from __future__ import annotations
 
-from collections.abc import Iterator
 from typing import (
     Any,
     TypeVar,
     Protocol,
     runtime_checkable,
+    TYPE_CHECKING,
 )
 
+if TYPE_CHECKING:
+    from collections.abc import Iterator
+
 __all__ = ["_NestedSequence"]
 
 _T_co = TypeVar("_T_co", covariant=True)

numpy/typing/mypy_plugin.py

@@ -33,11 +33,13 @@
 
 from __future__ import annotations
 
-from collections.abc import Iterable
 from typing import Final, TYPE_CHECKING, Callable
 
 import numpy as np
 
+if TYPE_CHECKING:
+    from collections.abc import Iterable
+
 try:
     import mypy.types
     from mypy.types import Type

numpy/typing/tests/data/pass/array_like.py

@@ -1,9 +1,11 @@
 from __future__ import annotations
 
-from typing import Any
+from typing import Any, TYPE_CHECKING
 
 import numpy as np
-from numpy._typing import NDArray, ArrayLike, _SupportsArray
+
+if TYPE_CHECKING:
+    from numpy._typing import NDArray, ArrayLike, _SupportsArray
 
 x1: ArrayLike = True
 x2: ArrayLike = 5

numpy/typing/tests/data/pass/literal.py

@@ -1,12 +1,14 @@
 from __future__ import annotations
 
-from typing import Any
+from typing import Any, TYPE_CHECKING
 from functools import partial
-from collections.abc import Callable
 
 import pytest
 import numpy as np
 
+if TYPE_CHECKING:
+    from collections.abc import Callable
+
 AR = np.array(0)
 AR.setflags(write=False)
 

numpy/typing/tests/test_typing.py

@@ -5,7 +5,6 @@
 import re
 import shutil
 from collections import defaultdict
-from collections.abc import Iterator
 from typing import TYPE_CHECKING
 
 import pytest
@@ -34,6 +33,7 @@
     NO_MYPY = False
 
 if TYPE_CHECKING:
+    from collections.abc import Iterator
     # We need this as annotation, but it's located in a private namespace.
     # As a compromise, do *not* import it during runtime
     from _pytest.mark.structures import ParameterSet