Register dict as potential device object

rapidsai · Mar 24, 2020 · 0a94220 · 0a94220
1 parent bf6f868
commit 0a94220
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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