Merge remote-tracking branch 'origin/master'

Author: nargokul

CHANGELOG.md

@@ -1,5 +1,30 @@
 # Changelog
 
+## v2.243.0 (2025-03-27)
+
+### Features
+
+ * Enabled update_endpoint through model_builder
+
+### Bug Fixes and Other Changes
+
+ * Update for PT 2.5.1, SMP 2.8.0
+ * chore: move jumpstart region definitions to json file
+ * fix flaky clarify model monitor test
+ * fix flaky spark processor integ
+ * use temp file in unit tests
+ * Update transformers version
+ * Aligned disable_output_compression for @remote with Estimator
+ * Update Jinja version
+ * update image_uri_configs  03-26-2025 07:18:16 PST
+ * chore: fix integ tests to use latest version of model
+ * update image_uri_configs  03-25-2025 07:18:13 PST
+ * Skip tests failed due to deprecated instance type
+ * update image_uri_configs  03-21-2025 07:17:55 PST
+ * factor in set instance type when building JumpStart models in ModelBuilder.
+ * ADD Documentation to ReadtheDocs for Upgrading torch versions
+ * add new regions to JUMPSTART_LAUNCHED_REGIONS
+
 ## v2.242.0 (2025-03-14)
 
 ### Features

VERSION

@@ -1 +1 @@
-2.242.1.dev0
+2.243.1.dev0

doc/amazon_sagemaker_model_building_pipeline.rst

@@ -408,21 +408,39 @@ Example:
         step_args=step_args_register_model,
     )
 
-CreateModelStep
+ModelStep
 ````````````````
 Referable Property List:
 
 - `DescribeModel`_
 
+  OR
+- `DescribeModelPackage`_
+
 .. _DescribeModel: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_DescribeModel.html#API_DescribeModel_ResponseSyntax
+.. _DescribeModelPackage: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_DescribeModelPackage.html#API_DescribeModelPackage_ResponseSyntax
 
 Example:
 
+For model creation usecase:
+
 .. code-block:: python
 
-    step_model = CreateModelStep(...)
-    model_data = step_model.PrimaryContainer.ModelDataUrl
+    create_model_step = ModelStep(
+        name="MyModelCreationStep",
+        step_args = model.create(...)
+    )
+    model_data = create_model_step.properties.PrimaryContainer.ModelDataUrl
+
+For model registration usercase:
+
+.. code-block:: python
 
+    register_model_step = ModelStep(
+        name="MyModelRegistrationStep",
+        step_args=model.register(...)
+    )
+    approval_status=register_model_step.properties.ModelApprovalStatus
 
 LambdaStep
 `````````````

doc/requirements.txt

@@ -2,7 +2,7 @@ sphinx==5.1.1
 sphinx-rtd-theme==0.5.0
 docutils==0.15.2
 packaging==20.9
-jinja2==3.1.4
+jinja2==3.1.6
 schema==0.7.5
 accelerate>=0.24.1,<=0.27.0
 graphene<4.0

requirements/extras/test_requirements.txt

@@ -20,7 +20,7 @@ attrs>=23.1.0,<24
 fabric==2.6.0
 requests==2.32.2
 sagemaker-experiments==0.1.35
-Jinja2==3.1.4
+Jinja2==3.1.6
 pyvis==0.2.1
 pandas==1.4.4
 scikit-learn==1.3.0
@@ -33,7 +33,7 @@ pillow>=10.0.1,<=11
 opentelemetry-proto==1.27.0
 protobuf==4.25.5
 tensorboard>=2.9.0,<=2.15.2
-transformers==4.46.1
+transformers==4.48.0
 sentencepiece==0.1.99
 # https://github.com/triton-inference-server/server/issues/6246
 tritonclient[http]<2.37.0

src/sagemaker/fw_utils.py

@@ -155,6 +155,7 @@
     "2.3.0",
     "2.3.1",
     "2.4.1",
+    "2.5.1",
 ]
 
 TRAINIUM_SUPPORTED_DISTRIBUTION_STRATEGIES = ["torch_distributed"]

src/sagemaker/huggingface/model.py

@@ -218,6 +218,7 @@ def deploy(
         container_startup_health_check_timeout=None,
         inference_recommendation_id=None,
         explainer_config=None,
+        update_endpoint: Optional[bool] = False,
         **kwargs,
     ):
         """Deploy this ``Model`` to an ``Endpoint`` and optionally return a ``Predictor``.
@@ -296,6 +297,11 @@ def deploy(
                 would like to deploy the model and endpoint with recommended parameters.
             explainer_config (sagemaker.explainer.ExplainerConfig): Specifies online explainability
                 configuration for use with Amazon SageMaker Clarify. (default: None)
+            update_endpoint (Optional[bool]):
+                Flag to update the model in an existing Amazon SageMaker endpoint.
+                If True, this will deploy a new EndpointConfig to an already existing endpoint
+                and delete resources corresponding to the previous EndpointConfig. Default: False
+                Note: Currently this is supported for single model endpoints
         Raises:
              ValueError: If arguments combination check failed in these circumstances:
                 - If no role is specified or
@@ -335,6 +341,7 @@ def deploy(
             container_startup_health_check_timeout=container_startup_health_check_timeout,
             inference_recommendation_id=inference_recommendation_id,
             explainer_config=explainer_config,
+            update_endpoint=update_endpoint,
             **kwargs,
         )