From 7068c63dd88d9faf3db09319945e5c52a1059770 Mon Sep 17 00:00:00 2001 From: Clark Zinzow Date: Thu, 3 Sep 2020 16:37:55 -0600 Subject: [PATCH] Set Ray task name to Dask key for Dask tasks. (#10547) --- python/ray/experimental/dask/scheduler.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/ray/experimental/dask/scheduler.py b/python/ray/experimental/dask/scheduler.py index b15164525..6d948cf25 100644 --- a/python/ray/experimental/dask/scheduler.py +++ b/python/ray/experimental/dask/scheduler.py @@ -196,7 +196,8 @@ def _rayify_task(task, key, deps): # Ray properly tracks the object dependencies between Ray tasks. object_refs, repack = unpack_object_refs(args, deps) # Submit the task using a wrapper function. - return dask_task_wrapper.remote(func, repack, *object_refs) + return dask_task_wrapper.options(name=f"dask:{key!s}").remote( + func, repack, *object_refs) elif not ishashable(task): return task elif task in deps: