FIX-modin-project#2254: Fixed dict aggregate for base backend

Signed-off-by: Gregory Shimansky <gregory.shimansky@intel.com>
gshimansky · Nov 10, 2020 · 9eea77c · 9eea77c
1 parent c1dc213
commit 9eea77c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/modin/data_management/functions/default_methods/groupby_default.py b/modin/data_management/functions/default_methods/groupby_default.py
@@ -80,7 +80,10 @@ def fn(
 
             grp = df.groupby(by, axis=axis, **groupby_args)
             agg_func = cls.get_func(grp, key, **kwargs)
-            result = agg_func(grp, **agg_args)
+            if isinstance(agg_func, dict):
+                result = grp.agg(agg_func, **agg_args)
+            else:
+                result = agg_func(grp, **agg_args)
 
             if not is_multi_by:
                 if as_index: