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):