Set Ray task name to Dask key for Dask tasks. (ray-project#10547)

erikerlandson · Sep 3, 2020 · 7068c63 · 7068c63
1 parent c54853d
commit 7068c63
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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: