diff --git a/cupy_xarray/accessors.py b/cupy_xarray/accessors.py index 325f2f3..06a5631 100644 --- a/cupy_xarray/accessors.py +++ b/cupy_xarray/accessors.py @@ -5,7 +5,10 @@ register_dataarray_accessor, register_dataset_accessor, ) -from xarray.core.pycompat import dask_array_type +from xarray.core.pycompat import DuckArrayModule + +dsk = DuckArrayModule("dask") +dask_array_type = dsk.type @register_dataarray_accessor("cupy")