From 12c558b7a22ee3a36fc2a5e18f829c8bfa6d240e Mon Sep 17 00:00:00 2001 From: Charles Blackmon-Luca <20627856+charlesbluca@users.noreply.github.com> Date: Tue, 10 May 2022 06:35:22 -0700 Subject: [PATCH] Add docstring to method decorator --- python/dask_cudf/dask_cudf/groupby.py | 6 ++++++ 1 file changed, 6 insertions(+) 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):