Add test cases for agg dicts with keys which are sub-types of Hashable and adjust type annotation to support this

Author: gandhis1

pandas-stubs/_typing.pyi

@@ -126,9 +126,9 @@ F = TypeVar("F", bound=FuncType)
 HashableT = TypeVar("HashableT", bound=Hashable)
 
 AggFuncTypeBase: TypeAlias = Union[Callable, str, np.ufunc]
-AggFuncTypeDictSeries: TypeAlias = Mapping[Hashable, AggFuncTypeBase]
+AggFuncTypeDictSeries: TypeAlias = Mapping[HashableT, AggFuncTypeBase]
 AggFuncTypeDictFrame: TypeAlias = Mapping[
-    Hashable, Union[AggFuncTypeBase, list[AggFuncTypeBase]]
+    HashableT, Union[AggFuncTypeBase, list[AggFuncTypeBase]]
 ]
 AggFuncTypeSeriesToFrame: TypeAlias = Union[
     list[AggFuncTypeBase],

tests/test_frame.py

@@ -670,6 +670,10 @@ def test_types_groupby_agg() -> None:
         0: lambda x: x.min(),
     }
     check(assert_type(df.groupby("col1").agg(agg_dict3), pd.DataFrame), pd.DataFrame)
+    agg_dict4 = {"col2": "sum"}
+    check(assert_type(df.groupby("col1").agg(agg_dict4), pd.DataFrame), pd.DataFrame)
+    agg_dict5 = {0: "sum"}
+    check(assert_type(df.groupby("col1").agg(agg_dict5), pd.DataFrame), pd.DataFrame)
     named_agg = pd.NamedAgg(column="col2", aggfunc="max")
     check(
         assert_type(df.groupby("col1").agg(new_col=named_agg), pd.DataFrame),