From f0ab2156e2ec023ea92db5a122488b16d76ed5f1 Mon Sep 17 00:00:00 2001 From: Zanie Blue Date: Fri, 23 Aug 2024 14:48:37 -0500 Subject: [PATCH] Set `VIRTUAL_ENV` for `uv run` invocations in ephemeral environments --- crates/uv/src/commands/project/run.rs | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/crates/uv/src/commands/project/run.rs b/crates/uv/src/commands/project/run.rs index 38964b63e04b..d4c8303e46ff 100644 --- a/crates/uv/src/commands/project/run.rs +++ b/crates/uv/src/commands/project/run.rs @@ -629,6 +629,11 @@ pub(crate) async fn run( )?; process.env("PATH", new_path); + // Ensure `VIRTUAL_ENV` is set. + if interpreter.is_virtualenv() { + process.env("VIRTUAL_ENV", interpreter.sys_prefix().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