missing cases

Author: cmp0xff

pandas-stubs/_typing.pyi

@@ -182,7 +182,7 @@ Level: TypeAlias = Hashable
 Shape: TypeAlias = tuple[int, ...]
 Suffixes: TypeAlias = tuple[str | None, str | None] | list[str | None]
 Ordered: TypeAlias = bool | None
-JSONSerializable: TypeAlias = PythonScalar | list[Any] | dict
+JSONSerializable: TypeAlias = PythonScalar | list[Any] | dict[str, Any]
 Frequency: TypeAlias = str | BaseOffset
 PeriodFrequency: TypeAlias = (
     str

pandas-stubs/core/reshape/pivot.pyi

@@ -35,7 +35,7 @@ from pandas._typing import (
 _PivotAggCallable: TypeAlias = Callable[[Series], ScalarT]
 
 _PivotAggFunc: TypeAlias = (
-    _PivotAggCallable
+    _PivotAggCallable[ScalarT]
     | np.ufunc
     | Literal["mean", "sum", "count", "min", "max", "median", "std", "var"]
 )
@@ -67,11 +67,17 @@ _Values: TypeAlias = SequenceNotStr[Any] | _ExtendedAnyArrayLike
 @overload
 def pivot_table(
     data: DataFrame,
-    values: _PivotTableValuesTypes = None,
-    index: _PivotTableIndexTypes = None,
-    columns: _PivotTableColumnsTypes = 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 | Sequence[_PivotAggFunc] | Mapping[Hashable, _PivotAggFunc]
+        _PivotAggFunc[Scalar]
+        | Sequence[_PivotAggFunc[Scalar]]
+        | Mapping[Hashable, _PivotAggFunc[Scalar]]
     ) = "mean",
     fill_value: Scalar | None = None,
     margins: bool = False,
@@ -85,12 +91,20 @@ def pivot_table(
 @overload
 def pivot_table(
     data: DataFrame,
-    values: _PivotTableValuesTypes = None,
+    values: _PivotTableValuesTypes[
+        Hashable  # ty: ignore[invalid-type-arguments]
+    ] = None,
     *,
     index: Grouper,
-    columns: _PivotTableColumnsTypes | np_ndarray | Index[Any] = None,
+    columns: (
+        _PivotTableColumnsTypes[Hashable]  # ty: ignore[invalid-type-arguments]
+        | np_ndarray
+        | Index[Any]
+    ) = None,
     aggfunc: (
-        _PivotAggFunc | Sequence[_PivotAggFunc] | Mapping[Hashable, _PivotAggFunc]
+        _PivotAggFunc[Scalar]
+        | Sequence[_PivotAggFunc[Scalar]]
+        | Mapping[Hashable, _PivotAggFunc[Scalar]]
     ) = "mean",
     fill_value: Scalar | None = None,
     margins: bool = False,
@@ -102,12 +116,20 @@ def pivot_table(
 @overload
 def pivot_table(
     data: DataFrame,
-    values: _PivotTableValuesTypes = None,
-    index: _PivotTableIndexTypes | 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: (
-        _PivotAggFunc | Sequence[_PivotAggFunc] | Mapping[Hashable, _PivotAggFunc]
+        _PivotAggFunc[Scalar]
+        | Sequence[_PivotAggFunc[Scalar]]
+        | Mapping[Hashable, _PivotAggFunc[Scalar]]
     ) = "mean",
     fill_value: Scalar | None = None,
     margins: bool = False,

pandas-stubs/core/series.pyi

@@ -776,7 +776,7 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
     def to_dict(self, *, into: type[dict] = ...) -> dict[Any, S1]: ...
     @overload
     def to_dict(
-        self, *, into: type[MutableMapping] | MutableMapping
+        self, *, into: type[MutableMapping] | MutableMapping[Hashable, Any]
     ) -> MutableMapping[Hashable, S1]: ...
     def to_frame(self, name: object | None = ...) -> DataFrame: ...
     @overload

pandas-stubs/core/tools/timedeltas.pyi

@@ -1,4 +1,3 @@
-from collections.abc import Sequence
 from datetime import timedelta
 from typing import overload
 
@@ -28,14 +27,7 @@ def to_timedelta(
 ) -> Series[Timedelta]: ...
 @overload
 def to_timedelta(
-    arg: (
-        SequenceNotStr
-        | Sequence[float | timedelta]
-        | tuple[str | float | timedelta, ...]
-        | range
-        | ArrayLike
-        | Index
-    ),
+    arg: SequenceNotStr[str | float | timedelta] | range | ArrayLike | Index,
     unit: TimeDeltaUnitChoices | None = ...,
     errors: RaiseCoerce = ...,
 ) -> TimedeltaIndex: ...

tests/test_frame.py

@@ -3991,10 +3991,10 @@ def test_to_dict_into_defaultdict() -> None:
     """Test DataFrame.to_dict with `into` is an instance of defaultdict[Any, list]"""
 
     data = pd.DataFrame({("str", "rts"): [[1, 2, 4], [2, 3], [3]]})
-    target: defaultdict[Any, list] = defaultdict(list)
+    target: defaultdict[Hashable, list[Any]] = defaultdict(list)
 
     check(
-        assert_type(data.to_dict(into=target), defaultdict[Any, list]),
+        assert_type(data.to_dict(into=target), defaultdict[Hashable, list[Any]]),
         defaultdict,
         tuple,
     )
@@ -4011,7 +4011,9 @@ def test_to_dict_into_defaultdict() -> None:
         str,
     )
     check(
-        assert_type(data.to_dict("records", into=target), list[defaultdict[Any, list]]),
+        assert_type(
+            data.to_dict("records", into=target), list[defaultdict[Hashable, list[Any]]]
+        ),
         list,
         defaultdict,
     )