Set VIRTUAL_ENV for uv run invocations in ephemeral environments

astral-sh · Aug 23, 2024 · 0c20825 · 0c20825
1 parent 9d1cd8e
commit 0c20825
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/crates/uv/src/commands/project/run.rs b/crates/uv/src/commands/project/run.rs
@@ -629,6 +629,10 @@ pub(crate) async fn run(
     )?;
     process.env("PATH", new_path);
 
+    if let Some(env) = ephemeral_env {
+        process.env("VIRTUAL_ENV", env.root().as_os_str());
+    };
+
     // Spawn and wait for completion
     // Standard input, output, and error streams are all inherited
     // TODO(zanieb): Throw a nicer error message if the command is not found