wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-22 13:00:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

[dask-on-ray] Convert tuple of object refs to list before ray.get() call. (#11582)

Browse Source
This commit is contained in:
Clark Zinzow
2020-10-23 16:39:22 -07:00
committed by GitHub
parent d3405e74da
commit 0979589c7c
2 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/util/dask/scheduler.py
+5 -3
View File
@@ -346,9 +346,11 @@ def ray_get_unpack(object_refs):
The input Python object with all contained Ray object references
resolved with their concrete values.
"""
if isinstance(object_refs,
(tuple, list)) and any(not isinstance(x, ray.ObjectRef)
for x in object_refs):
if isinstance(object_refs, tuple):
object_refs = list(object_refs)
if isinstance(object_refs, list) and any(not isinstance(x, ray.ObjectRef)
for x in object_refs):
# We flatten the object references before calling ray.get(), since Dask
# loves to nest collections in nested tuples and Ray expects a flat
# list of object references. We repack the results after ray.get()