check_untyped_defs pandas.core.groupby.groupby

Author: simonjayhawkins

pandas/core/groupby/groupby.py

@@ -12,7 +12,7 @@ class providing the base-class of operations.
 import datetime
 from functools import partial, wraps
 import types
-from typing import FrozenSet, List, Optional, Tuple, Type, Union
+from typing import Dict, FrozenSet, List, Optional, Tuple, Type, Union
 
 import numpy as np
 
@@ -849,7 +849,7 @@ def _transform_should_cast(self, func_nm):
         )
 
     def _cython_transform(self, how, numeric_only=True, **kwargs):
-        output = collections.OrderedDict()
+        output: Dict = collections.OrderedDict()
         for name, obj in self._iterate_slices():
             is_numeric = is_numeric_dtype(obj.dtype)
             if numeric_only and not is_numeric:
@@ -2259,7 +2259,7 @@ def _get_cythonized_result(
                 )
 
         labels, _, ngroups = grouper.group_info
-        output = collections.OrderedDict()
+        output: Dict = collections.OrderedDict()
         base_func = getattr(libgroupby, how)
 
         for name, obj in self._iterate_slices():
@@ -2502,7 +2502,8 @@ def _reindex_output(self, output):
 
 
 @Appender(GroupBy.__doc__)
-def groupby(obj, by, **kwds):
+def groupby(obj: Union[Series, DataFrame], by, **kwds) -> GroupBy:
+    klass: Type[GroupBy]
     if isinstance(obj, Series):
         from pandas.core.groupby.generic import SeriesGroupBy
 

setup.cfg

@@ -203,9 +203,6 @@ check_untyped_defs=False
 [mypy-pandas.core.groupby.generic]
 check_untyped_defs=False
 
-[mypy-pandas.core.groupby.groupby]
-check_untyped_defs=False
-
 [mypy-pandas.core.indexes.datetimes]
 check_untyped_defs=False