Merge branch 'main' of https://github.com/NVIDIA/NeMo-Skills

NVIDIA · Feb 16, 2025 · 7106de8 · 7106de8
2 parents 105b35e + 8b8b47f
commit 7106de8
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/nemo_skills/pipeline/utils.py b/nemo_skills/pipeline/utils.py
@@ -990,7 +990,15 @@ def add_task(
 
     # finally a sandbox if needed
     if with_sandbox:
-        with temporary_env_update(cluster_config, {"LISTEN_PORT": sandbox_port}):
+        sandbox_env_updates = {"LISTEN_PORT": sandbox_port}
+        current_env_vars = cluster_config.get("env_vars", []).copy()
+        for override in current_env_vars:
+            if "PYTHONPATH" in override:
+                if override.startswith("PYTHONPATH="):
+                    override = override[11:]
+                sandbox_env_updates["PYTHONPATH"] = override + ":/app"
+
+        with temporary_env_update(cluster_config, sandbox_env_updates):
             commands.append(get_sandox_command())
             sandbox_executor = get_executor(
                 cluster_config=cluster_config,