Apply suggestions from code review

Co-authored-by: nate nowack <thrast36@gmail.com>
PrefectHQ · Jan 8, 2025 · 23c4850 · 23c4850
1 parent 50ae925
commit 23c4850
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/prefect/flow_engine.py b/src/prefect/flow_engine.py
@@ -109,7 +109,7 @@ def load_flow_run(flow_run_id: UUID) -> FlowRun:
     return flow_run
 
 
-def load_flow(flow_run: FlowRun) -> Flow:
+def load_flow(flow_run: FlowRun) -> Flow[..., Any]:
     entrypoint = os.environ.get("PREFECT__FLOW_ENTRYPOINT")
 
     if entrypoint:
@@ -122,7 +122,7 @@ def load_flow(flow_run: FlowRun) -> Flow:
     return flow
 
 
-def load_flow_and_flow_run(flow_run_id: UUID) -> Tuple[FlowRun, Flow]:
+def load_flow_and_flow_run(flow_run_id: UUID) -> tuple[FlowRun, Flow[..., Any]]:
     flow_run = load_flow_run(flow_run_id)
     flow = load_flow(flow_run)
     return flow_run, flow