fix: add env to ensure_venv

bentoml · Sep 30, 2024 · 6a5a053 · 6a5a053
1 parent df8645c
commit 6a5a053
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/openllm/venv.py b/src/openllm/venv.py
@@ -13,7 +13,7 @@
 
 @functools.lru_cache
 def _resolve_bento_venv_spec(
-    bento: BentoInfo, runtime_envs: Optional[EnvVars] = None
+    bento: BentoInfo, runtime_envs: Optional[EnvVars] = None,
 ) -> VenvSpec:
     ver_file = bento.path / "env" / "python" / "version.txt"
     assert ver_file.exists(), f"cannot find version file in {bento.path}"
@@ -77,6 +77,7 @@ def _ensure_venv(venv_spec: VenvSpec) -> pathlib.Path:
                     venv / "requirements.txt",
                 ],
                 silent=VERBOSE_LEVEL.get() < 10,
+                env=venv_spec.envs,
             )
             with open(venv / "DONE", "w") as f:
                 f.write("DONE")