Enabled ruff TCH on some files (#51827)

* Enabled ruff TCH on pandas/core/apply.py, pandas/core/base.py, pandas/core/algorithms.py, pandas/core/sorting.py and pandas/core/ops/*. Only for the last two items there were changes needed. * black and isort fixed after running SKIP=flake8-rst pre-commit run --all-files --------- Co-authored-by: Marco Edward Gorelli <33491632+MarcoGorelli@users.noreply.github.com>

Author: ErdiTk

pandas/core/ops/array_ops.py

@@ -7,7 +7,10 @@
 import datetime
 from functools import partial
 import operator
-from typing import Any
+from typing import (
+ TYPE_CHECKING,
+ Any,
+)
 
 import numpy as np
 
@@ -19,10 +22,6 @@
  ops as libops,
 )
 from pandas._libs.tslibs import BaseOffset
-from pandas._typing import (
- ArrayLike,
- Shape,
-)
 
 from pandas.core.dtypes.cast import (
  construct_1d_object_array_from_listlike,
@@ -54,6 +53,12 @@
 from pandas.core.ops.dispatch import should_extension_dispatch
 from pandas.core.ops.invalid import invalid_comparison
 
+if TYPE_CHECKING:
+ from pandas._typing import (
+ ArrayLike,
+ Shape,
+ )
+
 # -----------------------------------------------------------------------------
 # Masking NA values and fallbacks for operations numpy does not support
 

pandas/core/ops/common.py

@@ -5,18 +5,23 @@
 
 from functools import wraps
 import sys
-from typing import Callable
+from typing import (
+ TYPE_CHECKING,
+ Callable,
+)
 
 from pandas._libs.lib import item_from_zerodim
 from pandas._libs.missing import is_matching_na
-from pandas._typing import F
 
 from pandas.core.dtypes.generic import (
  ABCDataFrame,
  ABCIndex,
  ABCSeries,
 )
 
+if TYPE_CHECKING:
+ from pandas._typing import F
+
 
 def unpack_zerodim_and_defer(name: str) -> Callable[[F], F]:
  """

pandas/core/ops/dispatch.py

@@ -3,12 +3,16 @@
 """
 from __future__ import annotations
 
-from typing import Any
-
-from pandas._typing import ArrayLike
+from typing import (
+ TYPE_CHECKING,
+ Any,
+)
 
 from pandas.core.dtypes.generic import ABCExtensionArray
 
+if TYPE_CHECKING:
+ from pandas._typing import ArrayLike
+
 
 def should_extension_dispatch(left: ArrayLike, right: Any) -> bool:
  """

pandas/core/sorting.py

@@ -20,15 +20,6 @@
  lib,
 )
 from pandas._libs.hashtable import unique_label_indices
-from pandas._typing import (
- AxisInt,
- IndexKeyFunc,
- Level,
- NaPosition,
- Shape,
- SortKind,
- npt,
-)
 
 from pandas.core.dtypes.common import (
  ensure_int64,
@@ -44,6 +35,16 @@
 from pandas.core.construction import extract_array
 
 if TYPE_CHECKING:
+ from pandas._typing import (
+ AxisInt,
+ IndexKeyFunc,
+ Level,
+ NaPosition,
+ Shape,
+ SortKind,
+ npt,
+ )
+
  from pandas import MultiIndex
  from pandas.core.arrays import ExtensionArray
  from pandas.core.indexes.base import Index

pyproject.toml

@@ -295,11 +295,6 @@ exclude = [
 # TCH to be enabled gradually
 "pandas/core/arrays/*" = ["TCH"]
 "pandas/core/nanops.py" = ["TCH"]
-"pandas/core/apply.py" = ["TCH"]
-"pandas/core/base.py" = ["TCH"]
-"pandas/core/algorithms.py" = ["TCH"]
-"pandas/core/ops/*" = ["TCH"]
-"pandas/core/sorting.py" = ["TCH"]
 "pandas/util/*" = ["TCH"]
 "pandas/_libs/*" = ["TCH"]
 # Keep this one enabled