Update python/dask_cudf/dask_cudf/backends.py

rapidsai · Aug 25, 2021 · f20463f · f20463f
1 parent 9b03113
commit f20463f
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/python/dask_cudf/dask_cudf/backends.py b/python/dask_cudf/dask_cudf/backends.py
@@ -260,13 +260,13 @@ def is_categorical_dtype_cudf(obj):
 
 try:
     try:
-        from dask.array.dispatch import (
-            percentile_lookup as percentile_dispatch,
-        )
+        from dask.array.dispatch import percentile_lookup
     except ImportError:
-        from dask.dataframe.dispatch import percentile_dispatch
+        from dask.dataframe.dispatch import (
+            percentile_dispatch as percentile_lookup,
+        )
 
-    @percentile_dispatch.register((cudf.Series, cp.ndarray, cudf.Index))
+    @percentile_lookup.register((cudf.Series, cp.ndarray, cudf.Index))
     def percentile_cudf(a, q, interpolation="linear"):
         # Cudf dispatch to the equivalent of `np.percentile`:
         # https://numpy.org/doc/stable/reference/generated/numpy.percentile.html