STYLE: Enable ruff TCH on some files (#51781)

STYLE enable ruff TCH

Co-authored-by: Marco Edward Gorelli <[email protected]>

Author: luke396

pandas/core/array_algos/masked_accumulations.py

@@ -5,18 +5,22 @@
 
 from __future__ import annotations
 
-from typing import Callable
+from typing import (
+    TYPE_CHECKING,
+    Callable,
+)
 
 import numpy as np
 
-from pandas._typing import npt
-
 from pandas.core.dtypes.common import (
     is_bool_dtype,
     is_float_dtype,
     is_integer_dtype,
 )
 
+if TYPE_CHECKING:
+    from pandas._typing import npt
+
 
 def _cum_func(
     func: Callable,

pandas/core/array_algos/masked_reductions.py

@@ -4,18 +4,23 @@
 """
 from __future__ import annotations
 
-from typing import Callable
+from typing import (
+    TYPE_CHECKING,
+    Callable,
+)
 
 import numpy as np
 
 from pandas._libs import missing as libmissing
-from pandas._typing import (
-    AxisInt,
-    npt,
-)
 
 from pandas.core.nanops import check_below_min_count
 
+if TYPE_CHECKING:
+    from pandas._typing import (
+        AxisInt,
+        npt,
+    )
+
 
 def _reductions(
     func: Callable,

pandas/core/array_algos/putmask.py

@@ -11,10 +11,6 @@
 import numpy as np
 
 from pandas._libs import lib
-from pandas._typing import (
-    ArrayLike,
-    npt,
-)
 from pandas.compat import np_version_under1p21
 
 from pandas.core.dtypes.cast import infer_dtype_from
@@ -23,6 +19,11 @@
 from pandas.core.arrays import ExtensionArray
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        ArrayLike,
+        npt,
+    )
+
     from pandas import MultiIndex
 
 

pandas/core/array_algos/quantile.py

@@ -1,19 +1,23 @@
 from __future__ import annotations
 
+from typing import TYPE_CHECKING
+
 import numpy as np
 
-from pandas._typing import (
-    ArrayLike,
-    Scalar,
-    npt,
-)
 from pandas.compat.numpy import np_percentile_argname
 
 from pandas.core.dtypes.missing import (
     isna,
     na_value_for_dtype,
 )
 
+if TYPE_CHECKING:
+    from pandas._typing import (
+        ArrayLike,
+        Scalar,
+        npt,
+    )
+
 
 def quantile_compat(
     values: ArrayLike, qs: npt.NDArray[np.float64], interpolation: str

pandas/core/array_algos/replace.py

@@ -6,25 +6,27 @@
 import operator
 import re
 from typing import (
+    TYPE_CHECKING,
     Any,
     Pattern,
 )
 
 import numpy as np
 
-from pandas._typing import (
-    ArrayLike,
-    Scalar,
-    npt,
-)
-
 from pandas.core.dtypes.common import (
     is_re,
     is_re_compilable,
     is_scalar,
 )
 from pandas.core.dtypes.missing import isna
 
+if TYPE_CHECKING:
+    from pandas._typing import (
+        ArrayLike,
+        Scalar,
+        npt,
+    )
+
 
 def should_use_regex(regex: bool, to_replace: Any) -> bool:
     """

pandas/core/array_algos/take.py

@@ -13,11 +13,6 @@
     algos as libalgos,
     lib,
 )
-from pandas._typing import (
-    ArrayLike,
-    AxisInt,
-    npt,
-)
 
 from pandas.core.dtypes.cast import maybe_promote
 from pandas.core.dtypes.common import (
@@ -29,6 +24,12 @@
 from pandas.core.construction import ensure_wrapped_if_datetimelike
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        ArrayLike,
+        AxisInt,
+        npt,
+    )
+
     from pandas.core.arrays._mixins import NDArrayBackedExtensionArray
     from pandas.core.arrays.base import ExtensionArray
 

pandas/core/array_algos/transforms.py

@@ -4,9 +4,12 @@
 
 from __future__ import annotations
 
+from typing import TYPE_CHECKING
+
 import numpy as np
 
-from pandas._typing import AxisInt
+if TYPE_CHECKING:
+    from pandas._typing import AxisInt
 
 
 def shift(values: np.ndarray, periods: int, axis: AxisInt, fill_value) -> np.ndarray:

pandas/core/dtypes/astype.py

@@ -15,11 +15,6 @@
 
 from pandas._libs import lib
 from pandas._libs.tslibs.timedeltas import array_to_timedelta64
-from pandas._typing import (
-    ArrayLike,
-    DtypeObj,
-    IgnoreRaise,
-)
 from pandas.errors import IntCastingNaNError
 
 from pandas.core.dtypes.common import (
@@ -37,8 +32,13 @@
 )
 
 if TYPE_CHECKING:
-    from pandas.core.arrays import ExtensionArray
+    from pandas._typing import (
+        ArrayLike,
+        DtypeObj,
+        IgnoreRaise,
+    )
 
+    from pandas.core.arrays import ExtensionArray
 
 _dtype_obj = np.dtype(object)
 

pandas/core/dtypes/base.py

@@ -15,12 +15,6 @@
 
 from pandas._libs import missing as libmissing
 from pandas._libs.hashtable import object_hash
-from pandas._typing import (
-    DtypeObj,
-    Shape,
-    npt,
-    type_t,
-)
 from pandas.errors import AbstractMethodError
 
 from pandas.core.dtypes.generic import (
@@ -30,6 +24,13 @@
 )
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        DtypeObj,
+        Shape,
+        npt,
+        type_t,
+    )
+
     from pandas.core.arrays import ExtensionArray
 
     # To parameterize on same ExtensionDtype

pandas/core/dtypes/cast.py

@@ -36,14 +36,6 @@
     is_supported_unit,
 )
 from pandas._libs.tslibs.timedeltas import array_to_timedelta64
-from pandas._typing import (
-    ArrayLike,
-    Dtype,
-    DtypeObj,
-    NumpyIndexT,
-    Scalar,
-    npt,
-)
 from pandas.errors import (
     IntCastingNaNError,
     LossySetitemError,
@@ -100,6 +92,15 @@
 )
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        ArrayLike,
+        Dtype,
+        DtypeObj,
+        NumpyIndexT,
+        Scalar,
+        npt,
+    )
+
     from pandas import Index
     from pandas.core.arrays import (
         Categorical,

pandas/core/dtypes/common.py

@@ -4,6 +4,7 @@
 from __future__ import annotations
 
 from typing import (
+    TYPE_CHECKING,
     Any,
     Callable,
 )
@@ -17,10 +18,6 @@
     lib,
 )
 from pandas._libs.tslibs import conversion
-from pandas._typing import (
-    ArrayLike,
-    DtypeObj,
-)
 
 from pandas.core.dtypes.base import _registry as registry
 from pandas.core.dtypes.dtypes import (
@@ -54,6 +51,12 @@
     is_sequence,
 )
 
+if TYPE_CHECKING:
+    from pandas._typing import (
+        ArrayLike,
+        DtypeObj,
+    )
+
 DT64NS_DTYPE = conversion.DT64NS_DTYPE
 TD64NS_DTYPE = conversion.TD64NS_DTYPE
 INT64_DTYPE = np.dtype(np.int64)

pandas/core/dtypes/concat.py

@@ -7,8 +7,6 @@
 
 import numpy as np
 
-from pandas._typing import AxisInt
-
 from pandas.core.dtypes.astype import astype_array
 from pandas.core.dtypes.cast import (
     common_dtype_categorical_compat,
@@ -26,6 +24,8 @@
 )
 
 if TYPE_CHECKING:
+    from pandas._typing import AxisInt
+
     from pandas.core.arrays import Categorical
 
 

pandas/core/dtypes/dtypes.py

@@ -31,13 +31,6 @@
     PeriodDtypeBase,
     abbrev_to_npy_unit,
 )
-from pandas._typing import (
-    Dtype,
-    DtypeObj,
-    Ordered,
-    npt,
-    type_t,
-)
 
 from pandas.core.dtypes.base import (
     ExtensionDtype,
@@ -57,6 +50,14 @@
 
     import pyarrow
 
+    from pandas._typing import (
+        Dtype,
+        DtypeObj,
+        Ordered,
+        npt,
+        type_t,
+    )
+
     from pandas import (
         Categorical,
         Index,