STYLE enable ruff TCH on pandas/core/arrays

pandas/core/arrays/_ranges.py

@@ -4,6 +4,8 @@
 """
 from __future__ import annotations
 
+from typing import TYPE_CHECKING
+
 import numpy as np
 
 from pandas._libs.lib import i8max
@@ -14,7 +16,9 @@
     Timestamp,
     iNaT,
 )
-from pandas._typing import npt
+
+if TYPE_CHECKING:
+    from pandas._typing import npt
 
 
 def generate_regular_range(

pandas/core/arrays/arrow/dtype.py

@@ -8,18 +8,14 @@
 )
 from decimal import Decimal
 import re
+from typing import TYPE_CHECKING
 
 import numpy as np
 
 from pandas._libs.tslibs import (
     Timedelta,
     Timestamp,
 )
-from pandas._typing import (
-    TYPE_CHECKING,
-    DtypeObj,
-    type_t,
-)
 from pandas.compat import pa_version_under7p0
 from pandas.util._decorators import cache_readonly
 
@@ -33,6 +29,11 @@
     import pyarrow as pa
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        DtypeObj,
+        type_t,
+    )
+
     from pandas.core.arrays.arrow import ArrowExtensionArray
 
 

pandas/core/arrays/arrow/extension_types.py

@@ -1,13 +1,15 @@
 from __future__ import annotations
 
 import json
+from typing import TYPE_CHECKING
 
 import pyarrow
 
-from pandas._typing import IntervalClosedType
-
 from pandas.core.arrays.interval import VALID_CLOSED
 
+if TYPE_CHECKING:
+    from pandas._typing import IntervalClosedType
+
 
 class ArrowPeriodType(pyarrow.ExtensionType):
     def __init__(self, freq) -> None:

pandas/core/arrays/base.py

@@ -25,20 +25,6 @@
 import numpy as np
 
 from pandas._libs import lib
-from pandas._typing import (
-    ArrayLike,
-    AstypeArg,
-    AxisInt,
-    Dtype,
-    FillnaOptions,
-    PositionalIndexer,
-    ScalarIndexer,
-    SequenceIndexer,
-    Shape,
-    SortKind,
-    TakeIndexer,
-    npt,
-)
 from pandas.compat import set_function_name
 from pandas.compat.numpy import function as nv
 from pandas.errors import AbstractMethodError
@@ -90,8 +76,20 @@
 
 if TYPE_CHECKING:
     from pandas._typing import (
+        ArrayLike,
+        AstypeArg,
+        AxisInt,
+        Dtype,
+        FillnaOptions,
         NumpySorter,
         NumpyValueArrayLike,
+        PositionalIndexer,
+        ScalarIndexer,
+        SequenceIndexer,
+        Shape,
+        SortKind,
+        TakeIndexer,
+        npt,
     )
 
 _extension_array_shared_docs: dict[str, str] = {}

pandas/core/arrays/boolean.py

@@ -12,11 +12,6 @@
     lib,
     missing as libmissing,
 )
-from pandas._typing import (
-    Dtype,
-    DtypeObj,
-    type_t,
-)
 
 from pandas.core.dtypes.common import (
     is_list_like,
@@ -35,7 +30,12 @@
 if TYPE_CHECKING:
     import pyarrow
 
-    from pandas._typing import npt
+    from pandas._typing import (
+        Dtype,
+        DtypeObj,
+        npt,
+        type_t,
+    )
 
 
 @register_extension_dtype

pandas/core/arrays/categorical.py

@@ -25,18 +25,6 @@
     lib,
 )
 from pandas._libs.arrays import NDArrayBacked
-from pandas._typing import (
-    ArrayLike,
-    AstypeArg,
-    AxisInt,
-    Dtype,
-    NpDtype,
-    Ordered,
-    Shape,
-    SortKind,
-    npt,
-    type_t,
-)
 from pandas.compat.numpy import function as nv
 from pandas.util._validators import validate_bool_kwarg
 
@@ -109,6 +97,19 @@
 from pandas.io.formats import console
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        ArrayLike,
+        AstypeArg,
+        AxisInt,
+        Dtype,
+        NpDtype,
+        Ordered,
+        Shape,
+        SortKind,
+        npt,
+        type_t,
+    )
+
     from pandas import (
         DataFrame,
         Index,

pandas/core/arrays/datetimes.py

@@ -42,13 +42,6 @@
     tzconversion,
 )
 from pandas._libs.tslibs.dtypes import abbrev_to_npy_unit
-from pandas._typing import (
-    DateTimeErrorChoices,
-    IntervalClosedType,
-    TimeAmbiguous,
-    TimeNonexistent,
-    npt,
-)
 from pandas.errors import PerformanceWarning
 from pandas.util._exceptions import find_stack_level
 from pandas.util._validators import validate_inclusive
@@ -87,6 +80,14 @@
 )
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        DateTimeErrorChoices,
+        IntervalClosedType,
+        TimeAmbiguous,
+        TimeNonexistent,
+        npt,
+    )
+
     from pandas import DataFrame
     from pandas.core.arrays import PeriodArray
 

pandas/core/arrays/numeric.py

@@ -15,11 +15,6 @@
     lib,
     missing as libmissing,
 )
-from pandas._typing import (
-    Dtype,
-    DtypeObj,
-    npt,
-)
 from pandas.errors import AbstractMethodError
 from pandas.util._decorators import cache_readonly
 
@@ -40,6 +35,12 @@
 if TYPE_CHECKING:
     import pyarrow
 
+    from pandas._typing import (
+        Dtype,
+        DtypeObj,
+        npt,
+    )
+
 
 T = TypeVar("T", bound="NumericArray")
 

pandas/core/arrays/numpy_.py

@@ -1,19 +1,14 @@
 from __future__ import annotations
 
+from typing import TYPE_CHECKING
+
 import numpy as np
 
 from pandas._libs import lib
 from pandas._libs.tslibs import (
     get_unit_from_dtype,
     is_supported_unit,
 )
-from pandas._typing import (
-    AxisInt,
-    Dtype,
-    NpDtype,
-    Scalar,
-    npt,
-)
 from pandas.compat.numpy import function as nv
 
 from pandas.core.dtypes.astype import astype_array
@@ -35,6 +30,15 @@
 from pandas.core.construction import ensure_wrapped_if_datetimelike
 from pandas.core.strings.object_array import ObjectStringArrayMixin
 
+if TYPE_CHECKING:
+    from pandas._typing import (
+        AxisInt,
+        Dtype,
+        NpDtype,
+        Scalar,
+        npt,
+    )
+
 
 class PandasArray(
     OpsMixin,

pandas/core/arrays/period.py

@@ -45,12 +45,6 @@
     get_period_field_arr,
     period_asfreq_arr,
 )
-from pandas._typing import (
-    AnyArrayLike,
-    Dtype,
-    NpDtype,
-    npt,
-)
 from pandas.util._decorators import (
     cache_readonly,
     doc,
@@ -81,8 +75,12 @@
 
 if TYPE_CHECKING:
     from pandas._typing import (
+        AnyArrayLike,
+        Dtype,
+        NpDtype,
         NumpySorter,
         NumpyValueArrayLike,
+        npt,
     )
 
     from pandas.core.arrays import (

pandas/core/arrays/sparse/array.py

@@ -28,19 +28,6 @@
     SparseIndex,
 )
 from pandas._libs.tslibs import NaT
-from pandas._typing import (
-    ArrayLike,
-    AstypeArg,
-    Axis,
-    AxisInt,
-    Dtype,
-    NpDtype,
-    PositionalIndexer,
-    Scalar,
-    ScalarIndexer,
-    SequenceIndexer,
-    npt,
-)
 from pandas.compat.numpy import function as nv
 from pandas.errors import PerformanceWarning
 from pandas.util._exceptions import find_stack_level
@@ -117,6 +104,20 @@ class ellipsis(Enum):
 
     SparseIndexKind = Literal["integer", "block"]
 
+    from pandas._typing import (
+        ArrayLike,
+        AstypeArg,
+        Axis,
+        AxisInt,
+        Dtype,
+        NpDtype,
+        PositionalIndexer,
+        Scalar,
+        ScalarIndexer,
+        SequenceIndexer,
+        npt,
+    )
+
     from pandas import Series
 
 else:

pandas/core/arrays/sparse/dtype.py

@@ -10,11 +10,6 @@
 
 import numpy as np
 
-from pandas._typing import (
-    Dtype,
-    DtypeObj,
-    type_t,
-)
 from pandas.errors import PerformanceWarning
 from pandas.util._exceptions import find_stack_level
 
@@ -36,6 +31,12 @@
 )
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        Dtype,
+        DtypeObj,
+        type_t,
+    )
+
     from pandas.core.arrays.sparse.array import SparseArray
 
 

pandas/core/arrays/sparse/scipy_sparse.py

@@ -10,13 +10,7 @@
     Iterable,
 )
 
-import numpy as np
-
 from pandas._libs import lib
-from pandas._typing import (
-    IndexLabel,
-    npt,
-)
 
 from pandas.core.dtypes.missing import notna
 
@@ -25,8 +19,14 @@
 from pandas.core.series import Series
 
 if TYPE_CHECKING:
+    import numpy as np
     import scipy.sparse
 
+    from pandas._typing import (
+        IndexLabel,
+        npt,
+    )
+
 
 def _check_is_partition(parts: Iterable, whole: Iterable):
     whole = set(whole)