Reraise if safe loading the flow fails

PrefectHQ · Jul 23, 2024 · 83327a3 · 83327a3
1 parent a3d3d81
commit 83327a3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/prefect/cli/deploy.py b/src/prefect/cli/deploy.py
@@ -472,7 +472,6 @@ async def _run_single_deploy(
         deploy_config["entrypoint"] = await prompt_entrypoint(app.console)
 
     flow = load_flow_from_entrypoint(deploy_config["entrypoint"])
-    assert flow is not None, "Flow could not be loaded from entrypoint."
 
     deploy_config["flow_name"] = flow.name
 

diff --git a/src/prefect/flows.py b/src/prefect/flows.py
@@ -1739,6 +1739,8 @@ def load_flow_from_entrypoint(
         # environment, fallback to loading the flow via AST parsing.
         if use_placeholder_flow:
             flow = safe_load_flow_from_entrypoint(entrypoint)
+            if flow is None:
+                raise
         else:
             raise