truss 0.11.9rc4__tar.gz → 0.11.9rc5__tar.gz

{truss-0.11.9rc4 → truss-0.11.9rc5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truss
-Version: 0.11.9rc4
+Version: 0.11.9rc5
 Summary: A seamless bridge from model development to model delivery
 Project-URL: Repository, https://github.com/basetenlabs/truss
 Project-URL: Homepage, https://truss.baseten.co

{truss-0.11.9rc4 → truss-0.11.9rc5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "truss"
-version = "0.11.9rc4"
+version = "0.11.9rc5"
 description = "A seamless bridge from model development to model delivery"
 authors = [
     { name = "Pankaj Gupta", email = "no-reply@baseten.co" },

{truss-0.11.9rc4 → truss-0.11.9rc5}/truss/templates/control/control/helpers/truss_patch/model_container_patch_applier.py

@@ -1,5 +1,6 @@
 import logging
 import os
+import shutil
 import subprocess
 from pathlib import Path
 from typing import Optional
@@ -21,6 +22,8 @@ from helpers.truss_patch.model_code_patch_applier import apply_code_patch
 from truss.base.truss_config import ExternalData, ExternalDataItem, TrussConfig
 from truss.util.download import download_external_data
 
+DEFAULT_PYTHON_EXECUTABLE = "python3"
+
 
 class ModelContainerPatchApplier:
     """Applies patches to container running a truss.
@@ -41,12 +44,19 @@ class ModelContainerPatchApplier:
             self._inference_server_home / ".." / self._truss_config.bundled_packages_dir
         ).resolve()
         self._data_dir = self._inference_server_home / self._truss_config.data_dir
-        self._python_executable = os.environ.get("PYTHON_EXECUTABLE", "python3")
         self._app_logger = app_logger
         self._uv_path_cached = None
         if uv_path is not None:
             self._uv_path_cached = uv_path
 
+        # NB(nikhil): Get full path to the python interpreter for pip patches.
+        self._python_executable = self._get_python_executable()
+
+    def _get_python_executable(self) -> str:
+        python_executable = os.environ.get("PYTHON_EXECUTABLE", "python3")
+        full_executable_path = shutil.which(python_executable)
+        return full_executable_path or python_executable
+
     def __call__(self, patch: Patch, inf_env: dict):
         self._app_logger.debug(f"Applying patch {patch.to_dict()}")
         if isinstance(patch.body, ModelCodePatch):