more convenient import

PrefectHQ · Feb 5, 2025 · d74512d · d74512d
1 parent cd2e4d3
commit d74512d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flows/update_flow_run_state_via_client.py b/flows/update_flow_run_state_via_client.py
@@ -9,7 +9,7 @@
 from uuid import UUID
 
 from prefect import flow, get_client
-from prefect.client.schemas import State, StateType
+from prefect.states import State, StateType
 
 
 def fail_anyway(flow_run_id: UUID):