STYLE: Enable ruff TCH on some files (pandas-dev#51812)

* STYLE: enable ruff TCH

* Fix circular import

Author: luke396

pandas/core/reshape/concat.py

@@ -19,11 +19,6 @@
 
 from pandas._config import using_copy_on_write
 
-from pandas._typing import (
-    Axis,
-    AxisInt,
-    HashableT,
-)
 from pandas.util._decorators import cache_readonly
 
 from pandas.core.dtypes.concat import concat_compat
@@ -51,6 +46,12 @@
 from pandas.core.internals import concatenate_managers
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        Axis,
+        AxisInt,
+        HashableT,
+    )
+
     from pandas import (
         DataFrame,
         Series,

pandas/core/reshape/encoding.py

@@ -3,14 +3,14 @@
 from collections import defaultdict
 import itertools
 from typing import (
+    TYPE_CHECKING,
     Hashable,
     Iterable,
 )
 
 import numpy as np
 
 from pandas._libs.sparse import IntIndex
-from pandas._typing import NpDtype
 
 from pandas.core.dtypes.common import (
     is_integer_dtype,
@@ -28,6 +28,9 @@
 )
 from pandas.core.series import Series
 
+if TYPE_CHECKING:
+    from pandas._typing import NpDtype
+
 
 def get_dummies(
     data,

pandas/core/reshape/pivot.py

@@ -11,12 +11,6 @@
 import numpy as np
 
 from pandas._libs import lib
-from pandas._typing import (
-    AggFuncType,
-    AggFuncTypeBase,
-    AggFuncTypeDict,
-    IndexLabel,
-)
 from pandas.util._decorators import (
     Appender,
     Substitution,
@@ -48,6 +42,13 @@
 from pandas.core.series import Series
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        AggFuncType,
+        AggFuncTypeBase,
+        AggFuncTypeDict,
+        IndexLabel,
+    )
+
     from pandas import DataFrame
 
 

pandas/core/reshape/reshape.py

@@ -10,7 +10,6 @@
 import numpy as np
 
 import pandas._libs.reshape as libreshape
-from pandas._typing import npt
 from pandas.errors import PerformanceWarning
 from pandas.util._decorators import cache_readonly
 from pandas.util._exceptions import find_stack_level
@@ -44,6 +43,8 @@
 )
 
 if TYPE_CHECKING:
+    from pandas._typing import npt
+
     from pandas.core.arrays import ExtensionArray
     from pandas.core.indexes.frozen import FrozenList
 

pandas/core/reshape/tile.py

@@ -4,6 +4,7 @@
 from __future__ import annotations
 
 from typing import (
+    TYPE_CHECKING,
     Any,
     Callable,
     Literal,
@@ -16,7 +17,6 @@
     Timestamp,
 )
 from pandas._libs.lib import infer_dtype
-from pandas._typing import IntervalLeftRight
 
 from pandas.core.dtypes.common import (
     DT64NS_DTYPE,
@@ -46,6 +46,9 @@
 from pandas.core import nanops
 import pandas.core.algorithms as algos
 
+if TYPE_CHECKING:
+    from pandas._typing import IntervalLeftRight
+
 
 def cut(
     x,

pandas/core/reshape/util.py

@@ -1,11 +1,14 @@
 from __future__ import annotations
 
-import numpy as np
+from typing import TYPE_CHECKING
 
-from pandas._typing import NumpyIndexT
+import numpy as np
 
 from pandas.core.dtypes.common import is_list_like
 
+if TYPE_CHECKING:
+    from pandas._typing import NumpyIndexT
+
 
 def cartesian_product(X) -> list[np.ndarray]:
     """

pandas/core/strings/base.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import abc
-import re
 from typing import (
     TYPE_CHECKING,
     Callable,
@@ -10,9 +9,11 @@
 
 import numpy as np
 
-from pandas._typing import Scalar
-
 if TYPE_CHECKING:
+    import re
+
+    from pandas._typing import Scalar
+
     from pandas import Series
 
 

pandas/core/strings/object_array.py

@@ -16,17 +16,18 @@
 from pandas._libs import lib
 import pandas._libs.missing as libmissing
 import pandas._libs.ops as libops
-from pandas._typing import (
-    NpDtype,
-    Scalar,
-)
 
 from pandas.core.dtypes.common import is_scalar
 from pandas.core.dtypes.missing import isna
 
 from pandas.core.strings.base import BaseStringArrayMethods
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        NpDtype,
+        Scalar,
+    )
+
     from pandas import Series
 
 

pandas/core/tools/numeric.py

@@ -1,6 +1,9 @@
 from __future__ import annotations
 
-from typing import Literal
+from typing import (
+    TYPE_CHECKING,
+    Literal,
+)
 
 import numpy as np
 
@@ -10,10 +13,6 @@
 )
 
 from pandas._libs import lib
-from pandas._typing import (
-    DateTimeErrorChoices,
-    npt,
-)
 
 from pandas.core.dtypes.cast import maybe_downcast_numeric
 from pandas.core.dtypes.common import (
@@ -36,6 +35,12 @@
 import pandas as pd
 from pandas.core.arrays import BaseMaskedArray
 
+if TYPE_CHECKING:
+    from pandas._typing import (
+        DateTimeErrorChoices,
+        npt,
+    )
+
 
 def to_numeric(
     arg,

pandas/core/tools/timedeltas.py

@@ -3,7 +3,6 @@
 """
 from __future__ import annotations
 
-from datetime import timedelta
 from typing import (
     TYPE_CHECKING,
     overload,
@@ -30,6 +29,8 @@
 from pandas.core.arrays.timedeltas import sequence_to_td64ns
 
 if TYPE_CHECKING:
+    from datetime import timedelta
+
     from pandas._libs.tslibs.timedeltas import UnitChoices
     from pandas._typing import (
         ArrayLike,

pandas/core/tools/times.py

@@ -4,18 +4,21 @@
     datetime,
     time,
 )
+from typing import TYPE_CHECKING
 
 import numpy as np
 
 from pandas._libs.lib import is_list_like
-from pandas._typing import DateTimeErrorChoices
 
 from pandas.core.dtypes.generic import (
     ABCIndex,
     ABCSeries,
 )
 from pandas.core.dtypes.missing import notna
 
+if TYPE_CHECKING:
+    from pandas._typing import DateTimeErrorChoices
+
 
 def to_time(
     arg,

pandas/core/window/ewm.py

@@ -9,15 +9,6 @@
 
 from pandas._libs.tslibs import Timedelta
 import pandas._libs.window.aggregations as window_aggregations
-from pandas._typing import (
-    Axis,
-    TimedeltaConvertibleTypes,
-)
-
-if TYPE_CHECKING:
-    from pandas import DataFrame, Series
-    from pandas.core.generic import NDFrame
-
 from pandas.util._decorators import doc
 
 from pandas.core.dtypes.common import (
@@ -60,6 +51,18 @@
     BaseWindowGroupby,
 )
 
+if TYPE_CHECKING:
+    from pandas._typing import (
+        Axis,
+        TimedeltaConvertibleTypes,
+    )
+
+    from pandas import (
+        DataFrame,
+        Series,
+    )
+    from pandas.core.generic import NDFrame
+
 
 def get_center_of_mass(
     comass: float | None,

pandas/core/window/expanding.py

@@ -7,16 +7,6 @@
     Callable,
 )
 
-from pandas._typing import (
-    Axis,
-    QuantileInterpolation,
-    WindowingRankType,
-)
-
-if TYPE_CHECKING:
-    from pandas import DataFrame, Series
-    from pandas.core.generic import NDFrame
-
 from pandas.util._decorators import doc
 
 from pandas.core.indexers.objects import (
@@ -40,6 +30,19 @@
     RollingAndExpandingMixin,
 )
 
+if TYPE_CHECKING:
+    from pandas._typing import (
+        Axis,
+        QuantileInterpolation,
+        WindowingRankType,
+    )
+
+    from pandas import (
+        DataFrame,
+        Series,
+    )
+    from pandas.core.generic import NDFrame
+
 
 class Expanding(RollingAndExpandingMixin):
     """

pandas/core/window/numba_.py

@@ -9,11 +9,13 @@
 
 import numpy as np
 
-from pandas._typing import Scalar
 from pandas.compat._optional import import_optional_dependency
 
 from pandas.core.util.numba_ import jit_user_function
 
+if TYPE_CHECKING:
+    from pandas._typing import Scalar
+
 
 @functools.lru_cache(maxsize=None)
 def generate_numba_apply_func(

pandas/core/window/rolling.py

@@ -26,13 +26,6 @@
     to_offset,
 )
 import pandas._libs.window.aggregations as window_aggregations
-from pandas._typing import (
-    ArrayLike,
-    Axis,
-    NDFrameT,
-    QuantileInterpolation,
-    WindowingRankType,
-)
 from pandas.compat._optional import import_optional_dependency
 from pandas.errors import DataError
 from pandas.util._decorators import doc
@@ -99,6 +92,14 @@
 )
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        ArrayLike,
+        Axis,
+        NDFrameT,
+        QuantileInterpolation,
+        WindowingRankType,
+    )
+
     from pandas import (
         DataFrame,
         Series,

pandas/io/__init__.py

@@ -1,3 +1,4 @@
+# ruff: noqa: TCH004
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
@@ -8,5 +9,5 @@
         stata,
     )
 
-    # and mark only those modules as public
+    # mark only those modules as public
     __all__ = ["formats", "json", "stata"]