From cc64ebcf9062920504bbb548b7e260d92155c7b7 Mon Sep 17 00:00:00 2001
From: Charles Blackmon-Luca <20627856+charlesbluca@users.noreply.github.com>
Date: Thu, 3 Nov 2022 11:44:48 -0700
Subject: [PATCH] Wrap groupby decorator with functools.wraps

---
 python/dask_cudf/dask_cudf/groupby.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/dask_cudf/dask_cudf/groupby.py b/python/dask_cudf/dask_cudf/groupby.py
index 7ae7651e689..54f8958c9eb 100644
--- a/python/dask_cudf/dask_cudf/groupby.py
+++ b/python/dask_cudf/dask_cudf/groupby.py
@@ -1,5 +1,6 @@
 # Copyright (c) 2020-2022, NVIDIA CORPORATION.
 
+from functools import wraps
 from typing import Set
 
 import numpy as np
@@ -38,6 +39,7 @@ def _check_groupby_optimized(func):
     reverting to the upstream Dask method
     """
 
+    @wraps(func)
     def wrapper(*args, **kwargs):
         gb = args[0]
         if _groupby_optimized(gb):