STY: remove --keep-runtime-typing from pyupgrade #40759 Part-1

Author: fathomer

.pre-commit-config.yaml

@@ -54,7 +54,7 @@ repos:
     rev: v2.11.0
     hooks:
     -   id: pyupgrade
-        args: [--py37-plus]
+        args: [--py37-plus, --keep-runtime-typing]
 -   repo: https://github.com/pre-commit/pygrep-hooks
     rev: v1.8.0
     hooks:

pandas/core/arrays/categorical.py

@@ -6,7 +6,10 @@
 from shutil import get_terminal_size
 from typing import (
     TYPE_CHECKING,
+    Dict,
     Hashable,
+    List,
+    Optional,
     Sequence,
     Tuple,
     Type,
@@ -356,7 +359,7 @@ def __init__(
         values,
         categories=None,
         ordered=None,
-        dtype: Dtype | None = None,
+        dtype: Optional[Dtype] = None,
         fastpath=False,
         copy: bool = True,
     ):
@@ -470,11 +473,11 @@ def dtype(self) -> CategoricalDtype:
         return self._dtype
 
     @property
-    def _constructor(self) -> type[Categorical]:
+    def _constructor(self) -> Type[Categorical]:
         return Categorical
 
     @classmethod
-    def _from_sequence(cls, scalars, *, dtype: Dtype | None = None, copy=False):
+    def _from_sequence(cls, scalars, *, dtype: Optional[Dtype] = None, copy=False):
         return Categorical(scalars, dtype=dtype, copy=copy)
 
     def astype(self, dtype: Dtype, copy: bool = True) -> ArrayLike:
@@ -544,7 +547,7 @@ def itemsize(self) -> int:
         """
         return self.categories.itemsize
 
-    def tolist(self) -> list[Scalar]:
+    def tolist(self) -> List[Scalar]:
         """
         Return a list of the values.
 
@@ -627,7 +630,7 @@ def _from_inferred_categories(
 
     @classmethod
     def from_codes(
-        cls, codes, categories=None, ordered=None, dtype: Dtype | None = None
+        cls, codes, categories=None, ordered=None, dtype: Optional[Dtype] = None
     ):
         """
         Make a Categorical type from codes and categories or dtype.
@@ -1366,7 +1369,7 @@ def _validate_fill_value(self, fill_value):
 
     # -------------------------------------------------------------
 
-    def __array__(self, dtype: NpDtype | None = None) -> np.ndarray:
+    def __array__(self, dtype: Optional[NpDtype] = None) -> np.ndarray:
         """
         The numpy array interface.
 
@@ -1957,7 +1960,7 @@ def _validate_setitem_value(self, value):
         codes = self.categories.get_indexer(rvalue)
         return codes.astype(self._ndarray.dtype, copy=False)
 
-    def _reverse_indexer(self) -> dict[Hashable, np.ndarray]:
+    def _reverse_indexer(self) -> Dict[Hashable, np.ndarray]:
         """
         Compute the inverse of a categorical, returning
         a dict of categories -> indexers.
@@ -2184,7 +2187,7 @@ def equals(self, other: object) -> bool:
 
     @classmethod
     def _concat_same_type(
-        cls: type[CategoricalT], to_concat: Sequence[CategoricalT], axis: int = 0
+        cls: Type[CategoricalT], to_concat: Sequence[CategoricalT], axis: int = 0
     ) -> CategoricalT:
         from pandas.core.dtypes.concat import union_categoricals
 

pandas/core/arrays/datetimelike.py

@@ -9,7 +9,10 @@
     TYPE_CHECKING,
     Any,
     Callable,
+    Optional,
     Sequence,
+    Tuple,
+    Type,
     TypeVar,
     Union,
     cast,
@@ -153,25 +156,25 @@ class DatetimeLikeArrayMixin(OpsMixin, NDArrayBackedExtensionArray):
     """
 
     # _infer_matches -> which infer_dtype strings are close enough to our own
-    _infer_matches: tuple[str, ...]
+    _infer_matches: Tuple[str, ...]
     _is_recognized_dtype: Callable[[DtypeObj], bool]
-    _recognized_scalars: tuple[type, ...]
+    _recognized_scalars: Tuple[Type, ...]
     _ndarray: np.ndarray
 
-    def __init__(self, data, dtype: Dtype | None = None, freq=None, copy=False):
+    def __init__(self, data, dtype: Optional[Dtype] = None, freq=None, copy=False):
         raise AbstractMethodError(self)
 
     @classmethod
     def _simple_new(
-        cls: type[DatetimeLikeArrayT],
+        cls: Type[DatetimeLikeArrayT],
         values: np.ndarray,
-        freq: BaseOffset | None = None,
-        dtype: Dtype | None = None,
+        freq: Optional[BaseOffset] = None,
+        dtype: Optional[Dtype] = None,
     ) -> DatetimeLikeArrayT:
         raise AbstractMethodError(cls)
 
     @property
-    def _scalar_type(self) -> type[DatetimeLikeScalar]:
+    def _scalar_type(self) -> Type[DatetimeLikeScalar]:
         """
         The scalar associated with this datelike
 
@@ -203,7 +206,7 @@ def _scalar_from_string(self, value: str) -> DTScalarOrNaT:
 
     def _unbox_scalar(
         self, value: DTScalarOrNaT, setitem: bool = False
-    ) -> np.int64 | np.datetime64 | np.timedelta64:
+    ) -> Union[np.int64, np.datetime64, np.timedelta64]:
         """
         Unbox the integer value of a scalar `value`.
 
@@ -331,15 +334,15 @@ def _formatter(self, boxed: bool = False):
     # ----------------------------------------------------------------
     # Array-Like / EA-Interface Methods
 
-    def __array__(self, dtype: NpDtype | None = None) -> np.ndarray:
+    def __array__(self, dtype: Optional[NpDtype] = None) -> np.ndarray:
         # used for Timedelta/DatetimeArray, overwritten by PeriodArray
         if is_object_dtype(dtype):
             return np.array(list(self), dtype=object)
         return self._ndarray
 
     def __getitem__(
-        self, key: int | slice | np.ndarray
-    ) -> DatetimeLikeArrayMixin | DTScalarOrNaT:
+        self, key: Union[int, slice, np.ndarray]
+    ) -> Union[DatetimeLikeArrayMixin, DTScalarOrNaT]:
         """
         This getitem defers to the underlying array, which by-definition can
         only handle list-likes, slices, and integer scalars
@@ -351,7 +354,7 @@ def __getitem__(
         result._freq = self._get_getitem_freq(key)
         return result
 
-    def _get_getitem_freq(self, key) -> BaseOffset | None:
+    def _get_getitem_freq(self, key) -> Optional[BaseOffset]:
         """
         Find the `freq` attribute to assign to the result of a __getitem__ lookup.
         """
@@ -383,8 +386,8 @@ def _get_getitem_freq(self, key) -> BaseOffset | None:
     # ndarray]"
     def __setitem__(  # type: ignore[override]
         self,
-        key: int | Sequence[int] | Sequence[bool] | slice,
-        value: NaTType | Any | Sequence[Any],
+        key: Union[int, Sequence[int], Sequence[bool], slice],
+        value: Union[NaTType, Any, Sequence[Any]],
     ) -> None:
         # I'm fudging the types a bit here. "Any" above really depends
         # on type(self). For PeriodArray, it's Period (or stuff coercible
@@ -466,10 +469,10 @@ def view(self, dtype: Literal["m8[ns]"]) -> TimedeltaArray:
         ...
 
     @overload
-    def view(self, dtype: Dtype | None = ...) -> ArrayLike:
+    def view(self, dtype: Optional[Dtype] = ...) -> ArrayLike:
         ...
 
-    def view(self, dtype: Dtype | None = None) -> ArrayLike:
+    def view(self, dtype: Optional[Dtype] = None) -> ArrayLike:
         # We handle datetime64, datetime64tz, timedelta64, and period
         #  dtypes here. Everything else we pass through to the underlying
         #  ndarray.
@@ -506,7 +509,7 @@ def view(self, dtype: Dtype | None = None) -> ArrayLike:
 
     @classmethod
     def _concat_same_type(
-        cls: type[DatetimeLikeArrayT],
+        cls: Type[DatetimeLikeArrayT],
         to_concat: Sequence[DatetimeLikeArrayT],
         axis: int = 0,
     ) -> DatetimeLikeArrayT:
@@ -542,7 +545,7 @@ def _values_for_factorize(self):
 
     @classmethod
     def _from_factorized(
-        cls: type[DatetimeLikeArrayT], values, original: DatetimeLikeArrayT
+        cls: Type[DatetimeLikeArrayT], values, original: DatetimeLikeArrayT
     ) -> DatetimeLikeArrayT:
         return cls(values, dtype=original.dtype)
 
@@ -786,7 +789,7 @@ def _validate_setitem_value(self, value):
 
     def _unbox(
         self, other, setitem: bool = False
-    ) -> np.int64 | np.datetime64 | np.timedelta64 | np.ndarray:
+    ) -> Union[np.int64, np.datetime64, np.timedelta64, np.ndarray]:
         """
         Unbox either a scalar with _unbox_scalar or an instance of our own type.
         """
@@ -936,7 +939,7 @@ def freq(self, value):
         self._freq = value
 
     @property
-    def freqstr(self) -> str | None:
+    def freqstr(self) -> Optional[str]:
         """
         Return the frequency object as a string if its set, otherwise None.
         """
@@ -945,7 +948,7 @@ def freqstr(self) -> str | None:
         return self.freq.freqstr
 
     @property  # NB: override with cache_readonly in immutable subclasses
-    def inferred_freq(self) -> str | None:
+    def inferred_freq(self) -> Optional[str]:
         """
         Tries to return a string representing a frequency guess,
         generated by infer_freq.  Returns None if it can't autodetect the
@@ -959,7 +962,7 @@ def inferred_freq(self) -> str | None:
             return None
 
     @property  # NB: override with cache_readonly in immutable subclasses
-    def _resolution_obj(self) -> Resolution | None:
+    def _resolution_obj(self) -> Optional[Resolution]:
         freqstr = self.freqstr
         if freqstr is None:
             return None
@@ -1017,7 +1020,7 @@ def _validate_frequency(cls, index, freq, **kwargs):
 
     @classmethod
     def _generate_range(
-        cls: type[DatetimeLikeArrayT], start, end, periods, freq, *args, **kwargs
+        cls: Type[DatetimeLikeArrayT], start, end, periods, freq, *args, **kwargs
     ) -> DatetimeLikeArrayT:
         raise AbstractMethodError(cls)
 
@@ -1440,7 +1443,7 @@ def __isub__(self, other):
     # --------------------------------------------------------------
     # Reductions
 
-    def min(self, *, axis: int | None = None, skipna: bool = True, **kwargs):
+    def min(self, *, axis: Optional[int] = None, skipna: bool = True, **kwargs):
         """
         Return the minimum value of the Array or minimum along
         an axis.
@@ -1469,7 +1472,7 @@ def min(self, *, axis: int | None = None, skipna: bool = True, **kwargs):
         result = nanops.nanmin(self._ndarray, axis=axis, skipna=skipna)
         return self._wrap_reduction_result(axis, result)
 
-    def max(self, *, axis: int | None = None, skipna: bool = True, **kwargs):
+    def max(self, *, axis: Optional[int] = None, skipna: bool = True, **kwargs):
         """
         Return the maximum value of the Array or maximum along
         an axis.
@@ -1500,7 +1503,7 @@ def max(self, *, axis: int | None = None, skipna: bool = True, **kwargs):
         result = nanops.nanmax(self._ndarray, axis=axis, skipna=skipna)
         return self._wrap_reduction_result(axis, result)
 
-    def mean(self, *, skipna: bool = True, axis: int | None = 0):
+    def mean(self, *, skipna: bool = True, axis: Optional[int] = 0):
         """
         Return the mean value of the Array.
 
@@ -1539,7 +1542,7 @@ def mean(self, *, skipna: bool = True, axis: int | None = 0):
         )
         return self._wrap_reduction_result(axis, result)
 
-    def median(self, *, axis: int | None = None, skipna: bool = True, **kwargs):
+    def median(self, *, axis: Optional[int] = None, skipna: bool = True, **kwargs):
         nv.validate_median((), kwargs)
 
         if axis is not None and abs(axis) >= self.ndim:
@@ -1749,11 +1752,11 @@ def ceil(self, freq, ambiguous="raise", nonexistent="raise"):
     # --------------------------------------------------------------
     # Reductions
 
-    def any(self, *, axis: int | None = None, skipna: bool = True):
+    def any(self, *, axis: Optional[int] = None, skipna: bool = True):
         # GH#34479 discussion of desired behavior long-term
         return nanops.nanany(self._ndarray, axis=axis, skipna=skipna, mask=self.isna())
 
-    def all(self, *, axis: int | None = None, skipna: bool = True):
+    def all(self, *, axis: Optional[int] = None, skipna: bool = True):
         # GH#34479 discussion of desired behavior long-term
         return nanops.nanall(self._ndarray, axis=axis, skipna=skipna, mask=self.isna())