truss 0.11.9rc3__py3-none-any.whl → 0.11.9rc5__py3-none-any.whl

truss/templates/control/control/helpers/truss_patch/model_container_patch_applier.py

@@ -1,4 +1,6 @@
 import logging
+import os
+import shutil
 import subprocess
 from pathlib import Path
 from typing import Optional
@@ -20,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.
@@ -45,6 +49,14 @@ class ModelContainerPatchApplier:
         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):
@@ -100,6 +112,8 @@ class ModelContainerPatchApplier:
                     "uninstall",
                     python_requirement_patch.requirement,
                     "--yes",
+                    "--python",
+                    self._python_executable,
                 ],
                 check=True,
             )
@@ -111,6 +125,8 @@ class ModelContainerPatchApplier:
                     "install",
                     python_requirement_patch.requirement,
                     "--upgrade",
+                    "--python",
+                    self._python_executable,
                 ],
                 check=True,
             )

{truss-0.11.9rc3.dist-info → truss-0.11.9rc5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truss
-Version: 0.11.9rc3
+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.9rc3.dist-info → truss-0.11.9rc5.dist-info}/RECORD

@@ -84,7 +84,7 @@ truss/templates/control/control/helpers/inference_server_process_controller.py,s
 truss/templates/control/control/helpers/inference_server_starter.py,sha256=Fz2Puijro6Cc5cvTsAqOveNJbBQR_ARYJXl4lvETJ8Y,2633
 truss/templates/control/control/helpers/truss_patch/__init__.py,sha256=CXZdUV_ylqLTJrKuFpvSnUT6PUFrZrMF2y6jiHbdaKU,998
 truss/templates/control/control/helpers/truss_patch/model_code_patch_applier.py,sha256=LTIIADLz0wRn7V49j64dU1U7Hbta9YLde3pb5YZWvzQ,2001
-truss/templates/control/control/helpers/truss_patch/model_container_patch_applier.py,sha256=FIn79RsVUtR6_uDF310aOm3tyIZjPNe2a3uOexzAdNI,6589
+truss/templates/control/control/helpers/truss_patch/model_container_patch_applier.py,sha256=VsOJYH40zv50bMDgaKe8HSRVju2kVaYou4NG25phKfE,7191
 truss/templates/control/control/helpers/truss_patch/requirement_name_identifier.py,sha256=CL3KEAj4B3ApMQShd7TI5umXVbazLZY5StrNlwHwWtc,1995
 truss/templates/control/control/helpers/truss_patch/system_packages.py,sha256=IYh1CVU_kooAvtSGXKQDDWnNdOhlv7ENWagsL1wvhgw,208
 truss/templates/custom/examples.yaml,sha256=2UcCtEdavImWmiCtj31ckBlAKVOwNMC5AwMIIznKDag,48
@@ -369,8 +369,8 @@ truss_train/deployment.py,sha256=lWWANSuzBWu2M4oK4qD7n-oVR1JKdmw2Pn5BJQHg-Ck,307
 truss_train/loader.py,sha256=0o66EjBaHc2YY4syxxHVR4ordJWs13lNXnKjKq2wq0U,1630
 truss_train/public_api.py,sha256=9N_NstiUlmBuLUwH_fNG_1x7OhGCytZLNvqKXBlStrM,1220
 truss_train/restore_from_checkpoint.py,sha256=8hdPm-WSgkt74HDPjvCjZMBpvA9MwtoYsxVjOoa7BaM,1176
-truss-0.11.9rc3.dist-info/METADATA,sha256=nKIkoDKhAZJh3mxQd1ePyp9WSAAe3VKhV7KcShXDU4Q,6680
-truss-0.11.9rc3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truss-0.11.9rc3.dist-info/entry_points.txt,sha256=-MwKfHHQHQ6j0HqIgvxrz3CehCmczDLTD-OsRHnjjuU,130
-truss-0.11.9rc3.dist-info/licenses/LICENSE,sha256=FTqGzu85i-uw1Gi8E_o0oD60bH9yQ_XIGtZbA1QUYiw,1064
-truss-0.11.9rc3.dist-info/RECORD,,
+truss-0.11.9rc5.dist-info/METADATA,sha256=DVCMx2eswWwbB5hv2xUPXB_JMgZt204AyQ41l8DRlOM,6680
+truss-0.11.9rc5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truss-0.11.9rc5.dist-info/entry_points.txt,sha256=-MwKfHHQHQ6j0HqIgvxrz3CehCmczDLTD-OsRHnjjuU,130
+truss-0.11.9rc5.dist-info/licenses/LICENSE,sha256=FTqGzu85i-uw1Gi8E_o0oD60bH9yQ_XIGtZbA1QUYiw,1064
+truss-0.11.9rc5.dist-info/RECORD,,