From 1cdaf8c1dedc0d225715a4efb723288a3b12e836 Mon Sep 17 00:00:00 2001 From: Justin Trautmann Date: Mon, 13 Nov 2023 15:21:21 +0100 Subject: [PATCH] options --- prefect_ray/task_runners.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/prefect_ray/task_runners.py b/prefect_ray/task_runners.py index 6013813..3ce4dde 100644 --- 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):