From 0a34f67c8e049728ddc62b997139d0fe8f122416 Mon Sep 17 00:00:00 2001
From: brandon-b-miller <53796099+brandon-b-miller@users.noreply.github.com>
Date: Fri, 27 May 2022 08:50:52 -0500
Subject: [PATCH] Register `cudf.core.groupby.Grouper` objects to dask
 `grouper_dispatch` (#10838) (#10982)

Backport of https://github.com/rapidsai/cudf/pull/10838

Authors:
   - https://github.com/brandon-b-miller

Approvers:
   - GALI PREM SAGAR (https://github.com/galipremsagar)
   - AJ Schmidt (https://github.com/ajschmidt8)
---
 python/dask_cudf/dask_cudf/backends.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/python/dask_cudf/dask_cudf/backends.py b/python/dask_cudf/dask_cudf/backends.py
index ac600c73285..1d3a15015fd 100644
--- a/python/dask_cudf/dask_cudf/backends.py
+++ b/python/dask_cudf/dask_cudf/backends.py
@@ -296,6 +296,16 @@ def is_categorical_dtype_cudf(obj):
     return cudf.api.types.is_categorical_dtype(obj)
 
 
+try:
+    from dask.dataframe.dispatch import grouper_dispatch
+
+    @grouper_dispatch.register((cudf.Series, cudf.DataFrame))
+    def get_grouper_cudf(obj):
+        return cudf.core.groupby.Grouper
+
+except ImportError:
+    pass
+
 try:
     try:
         from dask.array.dispatch import percentile_lookup