truss 0.11.2rc504__tar.gz → 0.11.3__tar.gz

{truss-0.11.2rc504 → truss-0.11.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truss
-Version: 0.11.2rc504
+Version: 0.11.3
 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.2rc504 → truss-0.11.3}/pyproject.toml

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

{truss-0.11.2rc504 → truss-0.11.3}/truss/api/__init__.py

@@ -64,6 +64,7 @@ def push(
     environment: Optional[str] = None,
     progress_bar: Optional[Type["progress.Progress"]] = None,
     include_git_info: bool = False,
+    preserve_env_instance_type: bool = True,
 ) -> definitions.ModelDeployment:
     """
     Pushes a Truss to Baseten.
@@ -88,6 +89,9 @@ def push(
         include_git_info: Whether to attach git versioning info (sha, branch, tag) to
           deployments made from within a git repo. If set to True in `.trussrc`, it
           will always be attached.
+        preserve_env_instance_type: When pushing a truss to an environment, whether to use the resources
+          specified in the truss config to resolve the instance type or preserve the instance type
+          configured in the specified environment.
 
     Returns:
         The newly created ModelDeployment.
@@ -130,6 +134,7 @@ def push(
         environment=environment,
         progress_bar=progress_bar,
         include_git_info=include_git_info,
+        preserve_env_instance_type=preserve_env_instance_type,
     )  # type: ignore
 
     return definitions.ModelDeployment(cast(BasetenService, service))

{truss-0.11.2rc504 → truss-0.11.3}/truss/templates/control/control/helpers/truss_patch/model_container_patch_applier.py

@@ -54,7 +54,9 @@ class ModelContainerPatchApplier:
             py_req_patch: PythonRequirementPatch = patch.body
             self._apply_python_requirement_patch(py_req_patch)
         elif isinstance(patch.body, SystemPackagePatch):
-            raise UnsupportedPatch("System package patches are not supported for model container, please push truss again instead")
+            raise UnsupportedPatch(
+                "System package patches are not supported for model container, please run truss push again"
+            )
         elif isinstance(patch.body, ConfigPatch):
             config_patch: ConfigPatch = patch.body
             self._apply_config_patch(config_patch)
@@ -113,7 +115,6 @@ class ModelContainerPatchApplier:
         else:
             raise ValueError(f"Unknown python requirement patch action {action}")
 
-
     def _apply_config_patch(self, config_patch: ConfigPatch):
         self._app_logger.debug(f"Applying config patch {config_patch.to_dict()}")
         TrussConfig.from_dict(config_patch.config).write_to_yaml_file(

{truss-0.11.2rc504 → truss-0.11.3}/truss/tests/patch/test_calc_patch.py

@@ -16,7 +16,6 @@ from truss.templates.control.control.helpers.custom_types import (
     Patch,
     PatchType,
     PythonRequirementPatch,
-    SystemPackagePatch,
 )
 from truss.truss_handle.patch.calc_patch import (
     _calc_python_requirements_patches,
@@ -736,49 +735,33 @@ def test_calc_config_patches_add_remove_env_var(custom_model_truss_dir: Path):
     ]
 
 
-def test_calc_config_patches_add_system_package(custom_model_truss_dir: Path):
-    patches = _apply_config_change_and_calc_patches(
-        custom_model_truss_dir, lambda config: config.system_packages.append("curl")
-    )
-    assert len(patches) == 2
-    assert patches == [
-        Patch(
-            type=PatchType.CONFIG,
-            body=ConfigPatch(
-                action=Action.UPDATE,
-                config=yaml.safe_load((custom_model_truss_dir / "config.yaml").open()),
-            ),
-        ),
-        Patch(
-            type=PatchType.SYSTEM_PACKAGE,
-            body=SystemPackagePatch(action=Action.ADD, package="curl"),
-        ),
-    ]
+def test_system_package_changes_make_truss_unpatchable_add(
+    custom_model_truss_dir: Path,
+):
+    # System package changes should raise a ValueError indicating full rebuild required
+    with pytest.raises(
+        ValueError, match="System package changes detected - full rebuild required"
+    ):
+        _apply_config_change_and_calc_patches(
+            custom_model_truss_dir, lambda config: config.system_packages.append("curl")
+        )
 
 
-def test_calc_config_patches_remove_system_package(custom_model_truss_dir: Path):
-    patches = _apply_config_change_and_calc_patches(
-        custom_model_truss_dir,
-        config_pre_op=lambda config: config.system_packages.append("curl"),
-        config_op=lambda config: config.system_packages.clear(),
-    )
-    assert len(patches) == 2
-    assert patches == [
-        Patch(
-            type=PatchType.CONFIG,
-            body=ConfigPatch(
-                action=Action.UPDATE,
-                config=yaml.safe_load((custom_model_truss_dir / "config.yaml").open()),
-            ),
-        ),
-        Patch(
-            type=PatchType.SYSTEM_PACKAGE,
-            body=SystemPackagePatch(action=Action.REMOVE, package="curl"),
-        ),
-    ]
+def test_system_package_changes_make_truss_unpatchable_remove(
+    custom_model_truss_dir: Path,
+):
+    # System package changes should raise a ValueError indicating full rebuild required
+    with pytest.raises(
+        ValueError, match="System package changes detected - full rebuild required"
+    ):
+        _apply_config_change_and_calc_patches(
+            custom_model_truss_dir,
+            config_pre_op=lambda config: config.system_packages.append("curl"),
+            config_op=lambda config: config.system_packages.clear(),
+        )
 
 
-def test_calc_config_patches_add_and_remove_system_package(
+def test_system_package_changes_make_truss_unpatchable_modify(
     custom_model_truss_dir: Path,
 ):
     def config_pre_op(config: TrussConfig):
@@ -787,29 +770,13 @@ def test_calc_config_patches_add_and_remove_system_package(
     def config_op(config: TrussConfig):
         config.system_packages = ["curl", "libsnd"]
 
-    patches = _apply_config_change_and_calc_patches(
-        custom_model_truss_dir, config_pre_op=config_pre_op, config_op=config_op
-    )
-    assert len(patches) == 3
-    assert patches[0] == Patch(
-        type=PatchType.CONFIG,
-        body=ConfigPatch(
-            action=Action.UPDATE,
-            config=yaml.safe_load((custom_model_truss_dir / "config.yaml").open()),
-        ),
-    )
-    patches = patches[1:]
-    patches.sort(key=lambda patch: patch.body.package)
-    assert patches == [
-        Patch(
-            type=PatchType.SYSTEM_PACKAGE,
-            body=SystemPackagePatch(action=Action.REMOVE, package="jq"),
-        ),
-        Patch(
-            type=PatchType.SYSTEM_PACKAGE,
-            body=SystemPackagePatch(action=Action.ADD, package="libsnd"),
-        ),
-    ]
+    # System package changes should raise a ValueError indicating full rebuild required
+    with pytest.raises(
+        ValueError, match="System package changes detected - full rebuild required"
+    ):
+        _apply_config_change_and_calc_patches(
+            custom_model_truss_dir, config_pre_op=config_pre_op, config_op=config_op
+        )
 
 
 def test_calc_config_patches_toggle_apply_library_patches(custom_model_truss_dir: Path):

{truss-0.11.2rc504 → truss-0.11.3}/truss/tests/test_control_truss_patching.py

@@ -9,8 +9,6 @@ from truss.tests.test_testing_utilities_for_other_tests import ensure_kill_all
 from truss.tests.test_truss_handle import (
     verify_python_requirement_installed_on_container,
     verify_python_requirement_not_installed_on_container,
-    verify_system_package_installed_on_container,
-    verify_system_requirement_not_installed_on_container,
 )
 from truss.truss_handle.truss_gatherer import calc_shadow_truss_dirname
 from truss.truss_handle.truss_handle import TrussHandle
@@ -147,14 +145,6 @@ def test_control_truss_python_sys_req_patch(
         th.remove_python_requirement(req)
         return th.docker_predict([1], tag=tag, binary=binary, local_port=None)
 
-    def predict_with_system_requirement_added(pkg):
-        th.add_system_package(pkg)
-        return th.docker_predict([1], tag=tag, binary=binary, local_port=None)
-
-    def predict_with_system_requirement_removed(pkg):
-        th.remove_system_package(pkg)
-        return th.docker_predict([1], tag=tag, binary=binary, local_port=None)
-
     with ensure_kill_all():
         th.docker_predict([1], tag=tag, binary=binary, local_port=None)
         orig_num_truss_images = len(th.get_all_docker_images())
@@ -170,15 +160,6 @@ def test_control_truss_python_sys_req_patch(
         assert current_num_docker_images(th) == orig_num_truss_images
         verify_python_requirement_not_installed_on_container(container, python_req)
 
-        system_pkg = "jq"
-        predict_with_system_requirement_added(system_pkg)
-        assert current_num_docker_images(th) == orig_num_truss_images
-        verify_system_package_installed_on_container(container, system_pkg)
-
-        predict_with_system_requirement_removed(system_pkg)
-        assert current_num_docker_images(th) == orig_num_truss_images
-        verify_system_requirement_not_installed_on_container(container, system_pkg)
-
 
 @pytest.mark.integration
 @pytest.mark.parametrize(

{truss-0.11.2rc504 → truss-0.11.3}/truss/truss_handle/patch/calc_patch.py

@@ -18,7 +18,6 @@ from truss.templates.control.control.helpers.custom_types import (
     Patch,
     PatchType,
     PythonRequirementPatch,
-    SystemPackagePatch,
 )
 from truss.templates.control.control.helpers.truss_patch.requirement_name_identifier import (
     RequirementMeta,
@@ -423,22 +422,18 @@ def _calc_python_requirements_patches(
 def _calc_system_packages_patches(
     prev_config: TrussConfig, new_config: TrussConfig
 ) -> List[Patch]:
-    """Calculate patch based on changes to system packates.
+    """Calculate patch based on changes to system packages.
 
-    Empty list means no relevant differences found.
+    System package patches are no longer supported, so this function
+    raises an exception if any system package changes are detected.
     """
-    patches = []
     prev_pkgs = system_packages_set(prev_config.system_packages)
     new_pkgs = system_packages_set(new_config.system_packages)
-    removed_pkgs = prev_pkgs.difference(new_pkgs)
-    for removed_pkg in removed_pkgs:
-        patches.append(_mk_system_package_patch(Action.REMOVE, removed_pkg))
 
-    added_pkgs = new_pkgs.difference(prev_pkgs)
-    for added_pkg in added_pkgs:
-        patches.append(_mk_system_package_patch(Action.ADD, added_pkg))
+    if prev_pkgs != new_pkgs:
+        raise ValueError("System package changes detected - full rebuild required")
 
-    return patches
+    return []
 
 
 def _mk_config_patch(action: Action, config: dict) -> Patch:
@@ -471,13 +466,6 @@ def _mk_python_requirement_patch(action: Action, requirement: str) -> Patch:
     )
 
 
-def _mk_system_package_patch(action: Action, package: str) -> Patch:
-    return Patch(
-        type=PatchType.SYSTEM_PACKAGE,
-        body=SystemPackagePatch(action=action, package=package),
-    )
-
-
 def _relative_to(path: str, relative_to_path: str):
     return str(Path(path).relative_to(relative_to_path))