diff --git a/python/dask_cudf/dask_cudf/groupby.py b/python/dask_cudf/dask_cudf/groupby.py index d666ec03a23..d137fac5fe3 100644 --- a/python/dask_cudf/dask_cudf/groupby.py +++ b/python/dask_cudf/dask_cudf/groupby.py @@ -36,6 +36,12 @@ def _check_groupby_supported(func): + """ + Decorator for dask-cudf's groupby methods that returns the dask-cudf + method if the groupby object is supported, otherwise reverting to the + upstream Dask method + """ + def wrapper(*args, **kwargs): gb = args[0] if _groupby_supported(gb):