Add kwargs to dask_cudf groupby.agg

rapidsai · Aug 12, 2024 · 3f3f6da · 3f3f6da
1 parent aa819c5
commit 3f3f6da
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/dask_cudf/dask_cudf/groupby.py b/python/dask_cudf/dask_cudf/groupby.py
@@ -196,7 +196,7 @@ def last(self, split_every=None, split_out=1):
     @_deprecate_shuffle_kwarg
     @_dask_cudf_performance_tracking
     def aggregate(
-        self, arg, split_every=None, split_out=1, shuffle_method=None
+        self, arg, split_every=None, split_out=1, shuffle_method=None, **kwargs
     ):
         if arg == "size":
             return self.size()
@@ -341,7 +341,7 @@ def last(self, split_every=None, split_out=1):
     @_deprecate_shuffle_kwarg
     @_dask_cudf_performance_tracking
     def aggregate(
-        self, arg, split_every=None, split_out=1, shuffle_method=None
+        self, arg, split_every=None, split_out=1, shuffle_method=None, **kwargs
     ):
         if arg == "size":
             return self.size()