STYLE: Enable ruff TCH on pandas/core/indexers and more (#51779)

enable ruff TCH

Author: luke396

pandas/core/groupby/generic.py

@@ -32,18 +32,6 @@
     lib,
     reduction as libreduction,
 )
-from pandas._typing import (
-    ArrayLike,
-    Axis,
-    AxisInt,
-    CorrelationMethod,
-    FillnaOptions,
-    IndexLabel,
-    Manager,
-    Manager2D,
-    SingleManager,
-    TakeIndexer,
-)
 from pandas.errors import SpecificationError
 from pandas.util._decorators import (
     Appender,
@@ -96,6 +84,19 @@
 from pandas.plotting import boxplot_frame_groupby
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        ArrayLike,
+        Axis,
+        AxisInt,
+        CorrelationMethod,
+        FillnaOptions,
+        IndexLabel,
+        Manager,
+        Manager2D,
+        SingleManager,
+        TakeIndexer,
+    )
+
     from pandas import Categorical
     from pandas.core.generic import NDFrame
 

pandas/core/groupby/grouper.py

@@ -17,12 +17,6 @@
 from pandas._config import using_copy_on_write
 
 from pandas._libs import lib
-from pandas._typing import (
-    ArrayLike,
-    Axis,
-    NDFrameT,
-    npt,
-)
 from pandas.errors import InvalidIndexError
 from pandas.util._decorators import cache_readonly
 from pandas.util._exceptions import find_stack_level
@@ -52,6 +46,13 @@
 from pandas.io.formats.printing import pprint_thing
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        ArrayLike,
+        Axis,
+        NDFrameT,
+        npt,
+    )
+
     from pandas.core.generic import NDFrame
 
 

pandas/core/groupby/indexing.py

@@ -9,7 +9,6 @@
 
 import numpy as np
 
-from pandas._typing import PositionalIndexer
 from pandas.util._decorators import (
     cache_readonly,
     doc,
@@ -21,6 +20,8 @@
 )
 
 if TYPE_CHECKING:
+    from pandas._typing import PositionalIndexer
+
     from pandas import (
         DataFrame,
         Series,

pandas/core/groupby/numba_.py

@@ -11,14 +11,16 @@
 
 import numpy as np
 
-from pandas._typing import Scalar
 from pandas.compat._optional import import_optional_dependency
 
 from pandas.core.util.numba_ import (
     NumbaUtilError,
     jit_user_function,
 )
 
+if TYPE_CHECKING:
+    from pandas._typing import Scalar
+
 
 def validate_udf(func: Callable) -> None:
     """

pandas/core/indexers/utils.py

@@ -10,8 +10,6 @@
 
 import numpy as np
 
-from pandas._typing import AnyArrayLike
-
 from pandas.core.dtypes.common import (
     is_array_like,
     is_bool_dtype,
@@ -26,6 +24,8 @@
 )
 
 if TYPE_CHECKING:
+    from pandas._typing import AnyArrayLike
+
     from pandas.core.frame import DataFrame
     from pandas.core.indexes.base import Index
 

pandas/core/indexes/api.py

@@ -1,15 +1,17 @@
 from __future__ import annotations
 
 import textwrap
-from typing import cast
+from typing import (
+    TYPE_CHECKING,
+    cast,
+)
 
 import numpy as np
 
 from pandas._libs import (
     NaT,
     lib,
 )
-from pandas._typing import Axis
 from pandas.errors import InvalidIndexError
 
 from pandas.core.dtypes.cast import find_common_type
@@ -30,6 +32,8 @@
 from pandas.core.indexes.range import RangeIndex
 from pandas.core.indexes.timedeltas import TimedeltaIndex
 
+if TYPE_CHECKING:
+    from pandas._typing import Axis
 _sort_msg = textwrap.dedent(
     """\
 Sorting because non-concatenation axis is not aligned. A future version

pandas/core/indexes/category.py

@@ -1,18 +1,14 @@
 from __future__ import annotations
 
 from typing import (
+    TYPE_CHECKING,
     Any,
     Hashable,
 )
 
 import numpy as np
 
 from pandas._libs import index as libindex
-from pandas._typing import (
-    Dtype,
-    DtypeObj,
-    npt,
-)
 from pandas.util._decorators import (
     cache_readonly,
     doc,
@@ -45,6 +41,12 @@
 
 from pandas.io.formats.printing import pprint_thing
 
+if TYPE_CHECKING:
+    from pandas._typing import (
+        Dtype,
+        DtypeObj,
+        npt,
+    )
 _index_doc_kwargs: dict[str, str] = dict(ibase._index_doc_kwargs)
 _index_doc_kwargs.update({"target_klass": "CategoricalIndex"})
 

pandas/core/indexes/datetimelike.py

@@ -7,7 +7,6 @@
     ABC,
     abstractmethod,
 )
-from datetime import datetime
 from typing import (
     TYPE_CHECKING,
     Any,
@@ -32,10 +31,6 @@
     parsing,
     to_offset,
 )
-from pandas._typing import (
-    Axis,
-    npt,
-)
 from pandas.compat.numpy import function as nv
 from pandas.errors import NullFrequencyError
 from pandas.util._decorators import (
@@ -70,6 +65,13 @@
 from pandas.core.tools.timedeltas import to_timedelta
 
 if TYPE_CHECKING:
+    from datetime import datetime
+
+    from pandas._typing import (
+        Axis,
+        npt,
+    )
+
     from pandas import CategoricalIndex
 
 _index_doc_kwargs = dict(ibase._index_doc_kwargs)

pandas/core/indexes/datetimes.py

@@ -25,15 +25,6 @@
     to_offset,
 )
 from pandas._libs.tslibs.offsets import prefix_mapping
-from pandas._typing import (
-    Dtype,
-    DtypeObj,
-    Frequency,
-    IntervalClosedType,
-    TimeAmbiguous,
-    TimeNonexistent,
-    npt,
-)
 from pandas.util._decorators import (
     cache_readonly,
     doc,
@@ -60,6 +51,16 @@
 from pandas.core.tools.times import to_time
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        Dtype,
+        DtypeObj,
+        Frequency,
+        IntervalClosedType,
+        TimeAmbiguous,
+        TimeNonexistent,
+        npt,
+    )
+
     from pandas.core.api import (
         DataFrame,
         PeriodIndex,

pandas/core/indexes/extension.py

@@ -11,10 +11,6 @@
 
 import numpy as np
 
-from pandas._typing import (
-    ArrayLike,
-    npt,
-)
 from pandas.util._decorators import (
     cache_readonly,
     doc,
@@ -25,6 +21,11 @@
 from pandas.core.indexes.base import Index
 
 if TYPE_CHECKING:
+    from pandas._typing import (
+        ArrayLike,
+        npt,
+    )
+
     from pandas.core.arrays import IntervalArray
     from pandas.core.arrays._mixins import NDArrayBackedExtensionArray
 

pandas/core/indexes/interval.py

@@ -7,6 +7,7 @@
 )
 import textwrap
 from typing import (
+    TYPE_CHECKING,
     Any,
     Hashable,
     Literal,
@@ -26,12 +27,6 @@
     Timestamp,
     to_offset,
 )
-from pandas._typing import (
-    Dtype,
-    DtypeObj,
-    IntervalClosedType,
-    npt,
-)
 from pandas.errors import InvalidIndexError
 from pandas.util._decorators import (
     Appender,
@@ -92,6 +87,13 @@
     timedelta_range,
 )
 
+if TYPE_CHECKING:
+    from pandas._typing import (
+        Dtype,
+        DtypeObj,
+        IntervalClosedType,
+        npt,
+    )
 _index_doc_kwargs = dict(ibase._index_doc_kwargs)
 
 _index_doc_kwargs.update(

pandas/core/indexes/period.py

@@ -4,7 +4,10 @@
     datetime,
     timedelta,
 )
-from typing import Hashable
+from typing import (
+    TYPE_CHECKING,
+    Hashable,
+)
 
 import numpy as np
 
@@ -16,11 +19,6 @@
     Resolution,
     Tick,
 )
-from pandas._typing import (
-    Dtype,
-    DtypeObj,
-    npt,
-)
 from pandas.util._decorators import (
     cache_readonly,
     doc,
@@ -46,6 +44,12 @@
 )
 from pandas.core.indexes.extension import inherit_names
 
+if TYPE_CHECKING:
+    from pandas._typing import (
+        Dtype,
+        DtypeObj,
+        npt,
+    )
 _index_doc_kwargs = dict(ibase._index_doc_kwargs)
 _index_doc_kwargs.update({"target_klass": "PeriodIndex or list of Periods"})
 _shared_doc_kwargs = {

pandas/core/indexes/range.py

@@ -4,6 +4,7 @@
 import operator
 from sys import getsizeof
 from typing import (
+    TYPE_CHECKING,
     Any,
     Callable,
     Hashable,
@@ -20,10 +21,6 @@
 )
 from pandas._libs.algos import unique_deltas
 from pandas._libs.lib import no_default
-from pandas._typing import (
-    Dtype,
-    npt,
-)
 from pandas.compat.numpy import function as nv
 from pandas.util._decorators import (
     cache_readonly,
@@ -51,6 +48,11 @@
 )
 from pandas.core.ops.common import unpack_zerodim_and_defer
 
+if TYPE_CHECKING:
+    from pandas._typing import (
+        Dtype,
+        npt,
+    )
 _empty_range = range(0)
 
 

pandas/core/indexes/timedeltas.py

@@ -1,6 +1,8 @@
 """ implement the TimedeltaIndex """
 from __future__ import annotations
 
+from typing import TYPE_CHECKING
+
 from pandas._libs import (
     index as libindex,
     lib,
@@ -10,7 +12,6 @@
     Timedelta,
     to_offset,
 )
-from pandas._typing import DtypeObj
 
 from pandas.core.dtypes.common import (
     is_dtype_equal,
@@ -28,6 +29,9 @@
 from pandas.core.indexes.datetimelike import DatetimeTimedeltaMixin
 from pandas.core.indexes.extension import inherit_names
 
+if TYPE_CHECKING:
+    from pandas._typing import DtypeObj
+
 
 @inherit_names(
     ["__neg__", "__pos__", "__abs__", "total_seconds", "round", "floor", "ceil"]