mypy, ty

Author: cmp0xff

pandas-stubs/_typing.pyi

@@ -963,9 +963,7 @@ ListLikeExceptSeriesAndStr: TypeAlias = (
     MutableSequence[Any] | np_1darray | tuple[Any, ...] | Index
 )
 ListLikeU: TypeAlias = Sequence | np_1darray | Series | Index
-ListLikeHashable: TypeAlias = (
-    MutableSequence[HashableT] | np_1darray | tuple[HashableT, ...] | range
-)
+ListLikeHashable: TypeAlias = SequenceNotStr[HashableT] | np_1darray | range
 
 class SupportsDType(Protocol[GenericT_co]):
     @property

pandas-stubs/core/frame.pyi

@@ -1174,7 +1174,7 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
     @overload
     def sort_values(
         self,
-        by: _str | Sequence[_str],
+        by: _str | SequenceNotStr[_str],
         *,
         axis: Axis = 0,
         ascending: _bool | Sequence[_bool] = ...,
@@ -1187,7 +1187,7 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
     @overload
     def sort_values(
         self,
-        by: _str | Sequence[_str],
+        by: _str | SequenceNotStr[_str],
         *,
         axis: Axis = 0,
         ascending: _bool | Sequence[_bool] = ...,

pandas-stubs/core/groupby/generic.pyi

@@ -261,7 +261,7 @@ class DataFrameGroupBy(GroupBy[DataFrame], Generic[ByT, _TT]):
     @overload
     def aggregate(
         self,
-        func: AggFuncTypeFrame[Hashable] | None = ...,
+        func: AggFuncTypeFrame[Any] | None = ...,
         *args: Any,
         engine: WindowingEngine = ...,
         engine_kwargs: WindowingEngineKwargs = ...,
@@ -270,7 +270,7 @@ class DataFrameGroupBy(GroupBy[DataFrame], Generic[ByT, _TT]):
     @overload
     def aggregate(
         self,
-        func: AggFuncTypeFrame[Hashable] | None = None,
+        func: AggFuncTypeFrame[Any] | None = None,
         /,
         **kwargs: Any,
     ) -> DataFrame: ...

pandas-stubs/core/reshape/pivot.pyi

@@ -66,9 +66,13 @@ _ExtendedAnyArrayLike: TypeAlias = AnyArrayLike | ArrayLike
 @overload
 def pivot_table(
     data: DataFrame,
-    values: _PivotTableValuesTypes[Hashable] = None,
-    index: _PivotTableIndexTypes[Hashable] = None,
-    columns: _PivotTableColumnsTypes[Hashable] = None,
+    values: _PivotTableValuesTypes[
+        Hashable  # ty: ignore[invalid-type-arguments]
+    ] = None,
+    index: _PivotTableIndexTypes[Hashable] = None,  # ty: ignore[invalid-type-arguments]
+    columns: _PivotTableColumnsTypes[
+        Hashable  # ty: ignore[invalid-type-arguments]
+    ] = None,
     aggfunc: (
         _PivotAggFunc[Any]
         | Sequence[_PivotAggFunc[Any]]
@@ -86,10 +90,16 @@ def pivot_table(
 @overload
 def pivot_table(
     data: DataFrame,
-    values: _PivotTableValuesTypes[Hashable] = None,
+    values: _PivotTableValuesTypes[
+        Hashable  # ty: ignore[invalid-type-arguments]
+    ] = None,
     *,
     index: Grouper,
-    columns: _PivotTableColumnsTypes[Hashable] | np_ndarray | Index[Any] = None,
+    columns: (
+        _PivotTableColumnsTypes[Hashable]  # ty: ignore[invalid-type-arguments]
+        | np_ndarray
+        | Index[Any]
+    ) = None,
     aggfunc: (
         _PivotAggFunc[Any]
         | Sequence[_PivotAggFunc[Any]]
@@ -105,8 +115,14 @@ def pivot_table(
 @overload
 def pivot_table(
     data: DataFrame,
-    values: _PivotTableValuesTypes[Hashable] = None,
-    index: _PivotTableIndexTypes[Hashable] | np_ndarray | Index[Any] = None,
+    values: _PivotTableValuesTypes[
+        Hashable  # ty: ignore[invalid-type-arguments]
+    ] = None,
+    index: (
+        _PivotTableIndexTypes[Hashable]  # ty: ignore[invalid-type-arguments]
+        | np_ndarray
+        | Index[Any]
+    ) = None,
     *,
     columns: Grouper,
     aggfunc: (