Fix dependency installation for flow submission (#17289)

PrefectHQ · Feb 26, 2025 · 27db9ba · 27db9ba
1 parent 6836cce
commit 27db9ba
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/prefect/_experimental/bundles.py b/src/prefect/_experimental/bundles.py
@@ -149,6 +149,8 @@ def execute_bundle_in_subprocess(
     if dependencies := bundle.get("dependencies"):
         subprocess.check_call(
             [uv.find_uv_bin(), "pip", "install", *dependencies.split("\n")],
+            # Copy the current environment to ensure we install into the correct venv
+            env=os.environ,
         )
 
     process = ctx.Process(

diff --git a/tests/experimental/test_bundles.py b/tests/experimental/test_bundles.py
@@ -1,3 +1,4 @@
+import os
 import signal
 import subprocess
 import sys
@@ -76,7 +77,8 @@ async def simple_flow():
                 "pip",
                 "install",
                 "the-whole-enchilada==0.5.3",
-            ]
+            ],
+            env=os.environ,
         )
 
         flow_run = await prefect_client.read_flow_run(flow_run.id)