options

PrefectHQ · Nov 13, 2023 · 1cdaf8c · 1cdaf8c
1 parent b67a5a3
commit 1cdaf8c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/prefect_ray/task_runners.py b/prefect_ray/task_runners.py
@@ -194,10 +194,10 @@ def resolve_ray_future(expr):
 
             return func(**kwargs)
 
-        _run_prefect_task.__qualname__ = call.keywords["task_run"].name
-
-        self._ray_refs[key] = ray_decorator(_run_prefect_task).remote(
-            sync_compatible(call.func), *upstream_ray_obj_refs, **call_kwargs
+        self._ray_refs[key] = (
+            ray_decorator(_run_prefect_task)
+            .options(call.keywords["task_run"].name)
+            .remote(sync_compatible(call.func), *upstream_ray_obj_refs, **call_kwargs)
         )
 
     def _exchange_prefect_for_ray_futures(self, kwargs_prefect_futures):