FIX-modin-project#2254: Lint fixes

Signed-off-by: Gregory Shimansky <gregory.shimansky@intel.com>

modin/backends/pandas/query_compiler.py

@@ -2579,7 +2579,11 @@ def groupby_agg(
             # because default to pandas is unable to operate with dictionary aggregation function argument,
             # it accepts only callable functions.
             if isinstance(agg_func, dict):
-                callable_func = lambda df, *args, **kwargs: df.aggregate(agg_func, *agg_args, **agg_kwargs)
+                callable_func = wrap_udf_function(
+                    lambda df, *args, **kwargs: df.aggregate(
+                        agg_func, *agg_args, **agg_kwargs
+                    )
+                )
             else:
                 callable_func = agg_func
             return super().groupby_agg(

modin/pandas/groupby.py

@@ -873,7 +873,9 @@ def _apply_agg_function(self, f, drop=True, *args, **kwargs):
         -------
         A new combined DataFrame with the result of all groups.
         """
-        assert callable(f) or isinstance(f, dict), "'{0}' object is not callable and not a dict".format(type(f))
+        assert callable(f) or isinstance(
+            f, dict
+        ), "'{0}' object is not callable and not a dict".format(type(f))
 
         # For aggregations, pandas behavior does this for the result.
         # For other operations it does not, so we wait until there is an aggregation to