From fc8bd37bc0398240c88d0480440d9264e294c211 Mon Sep 17 00:00:00 2001 From: Alexander Streed Date: Fri, 26 Apr 2024 11:28:47 -0500 Subject: [PATCH] Update task_runners.py --- prefect_ray/task_runners.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/prefect_ray/task_runners.py b/prefect_ray/task_runners.py index 39a79a1..b73d0fd 100644 --- a/prefect_ray/task_runners.py +++ b/prefect_ray/task_runners.py @@ -78,7 +78,7 @@ def count_to(highest_number): import anyio import ray from prefect.futures import PrefectFuture -from prefect.server.schemas.states import State +from prefect.client.schemas.objects import State from prefect.states import exception_to_crashed_state from prefect.task_runners import BaseTaskRunner, R, TaskConcurrencyType from prefect.utilities.asyncutils import sync_compatible