diff --git a/dask_cuda/is_device_object.py b/dask_cuda/is_device_object.py index f11a6cdee..0654b4b4d 100644 --- a/dask_cuda/is_device_object.py +++ b/dask_cuda/is_device_object.py @@ -18,6 +18,11 @@ def is_device_object_python_collection(seq): return any([is_device_object(s) for s in seq]) +@is_device_object.register(dict) +def is_device_object_python_dict(seq): + return any([is_device_object(s) for s in seq.items()]) + + @is_device_object.register_lazy("cudf") def register_cudf(): import cudf