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

2026-07-01 02:49:33 +08:00 · 2020-09-03 16:37:55 -06:00
parent c54853d45b
commit 7068c63dd8
1 changed files with 2 additions and 1 deletions
@@ -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: