From 0a942204eda195587f4a6aef505f6265c1d5b214 Mon Sep 17 00:00:00 2001 From: "Mads R. B. Kristensen" Date: Tue, 24 Mar 2020 11:11:11 -0400 Subject: [PATCH] Register dict as potential device object --- dask_cuda/is_device_object.py | 5 +++++ 1 file changed, 5 insertions(+) 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