Merge branch 'master' into master

zhaoqizqwang · web-flow · commit 7f8a705c0046 · 2026-01-14T10:40:48.000-08:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,13 @@
 # Changelog
 
+## v3.3.1 (2026-01-12)
+### Bug fixes and Other Changes
+  * ProcessingJob fix - Remove tags in Processor while Job creation
+  * Telemetry Updates
+  * sagemaker-mlops bug fix - Correct source code 'dependencies' parameter to 'requirements'
+  * aws_batch bug fix - remove experiment config parameter as it Estimator is deprecated.
+
+
 ## v3.3.0 (2025-12-19)
 
 ### Features
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-3.3.0
+3.3.1
diff --git a/pyproject.toml b/pyproject.toml
@@ -33,10 +33,10 @@ classifiers = [
   "Programming Language :: Python :: 3.12",
 ]
 dependencies = [
-  "sagemaker-core>=2.3.0,<3.0.0",
-  "sagemaker-train>=1.3.0,<2.0.0",
-  "sagemaker-serve>=1.3.0,<2.0.0",
-  "sagemaker-mlops>=1.3.0,<2.0.0",
+  "sagemaker-core>=2.3.1,<3.0.0",
+  "sagemaker-train>=1.3.1,<2.0.0",
+  "sagemaker-serve>=1.3.1,<2.0.0",
+  "sagemaker-mlops>=1.3.1,<2.0.0",
 ]
 
 [project.optional-dependencies]
diff --git a/sagemaker-core/CHANGELOG.md b/sagemaker-core/CHANGELOG.md
@@ -1,5 +1,10 @@
 # Changelog
 
+## v2.3.1 (2026-01-12)
+
+### Bug Fixes
+* ProcessingJob fix - Remove tags in Processor while Job creation
+
 ## v2.3.0 (2025-12-19)
 
 ### Feature
diff --git a/sagemaker-core/VERSION b/sagemaker-core/VERSION
@@ -1,2 +1,2 @@
-2.3.0
+2.3.1
 
diff --git a/sagemaker-core/src/sagemaker/core/jumpstart/models.py b/sagemaker-core/src/sagemaker/core/jumpstart/models.py
@@ -129,6 +129,7 @@ class InferenceAmiVersionEnum(StrEnum):
     """InferenceAmiVersionEnum"""
 
     AL2_AMI_SAGEMAKER_INFERENCE_GPU_2 = "al2-ami-sagemaker-inference-gpu-2"
+    AL2_AMI_SAGEMAKER_INFERENCE_GPU_3_1 = "al2-ami-sagemaker-inference-gpu-3-1"
 
 
 class ScopeEnum(StrEnum):
diff --git a/sagemaker-core/src/sagemaker/core/workflow/utilities.py b/sagemaker-core/src/sagemaker/core/workflow/utilities.py
@@ -172,9 +172,9 @@ def get_code_hash(step: Entity) -> str:
         source_code = model_trainer.source_code
         if source_code:
             source_dir = source_code.source_dir
-            dependencies = source_code.dependencies
+            requirements = source_code.requirements
             entry_point = source_code.entry_script
-            return get_training_code_hash(entry_point, source_dir, dependencies)
+            return get_training_code_hash(entry_point, source_dir, requirements)
     return None
 
 
diff --git a/sagemaker-mlops/CHANGELOG.md b/sagemaker-mlops/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Changelog
 
+## v1.3.1 (2026-01-12)
+
+* sagemaker-mlops bug fix - Correct source code 'dependencies' parameter to 'requirements'
+  
 ## v1.3.0 (2025-12-19)
 
 * Update module dependencies
diff --git a/sagemaker-mlops/VERSION b/sagemaker-mlops/VERSION
@@ -1 +1 @@
-1.3.0
+1.3.1
diff --git a/sagemaker-mlops/pyproject.toml b/sagemaker-mlops/pyproject.toml
@@ -22,9 +22,9 @@ classifiers = [
     "Programming Language :: Python :: 3.12",
 ]
 dependencies = [
-    "sagemaker-core>=2.3.0",
-    "sagemaker-train>=1.3.0",
-    "sagemaker-serve>=1.3.0",
+    "sagemaker-core>=2.3.1",
+    "sagemaker-train>=1.3.1",
+    "sagemaker-serve>=1.3.1",
     "boto3>=1.42.2,<2.0",
     "botocore>=1.42.2,<2.0",
 ]
diff --git a/sagemaker-serve/CHANGELOG.md b/sagemaker-serve/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Changelog
 
+## v1.3.1 (2026-01-12)
+
+* Update Module dependencies
+
 ## v1.3.0 (2025-12-19)
 
 ### Bug Fixes
diff --git a/sagemaker-serve/VERSION b/sagemaker-serve/VERSION
@@ -1 +1 @@
-1.3.0
+1.3.1
diff --git a/sagemaker-serve/pyproject.toml b/sagemaker-serve/pyproject.toml
@@ -22,8 +22,8 @@ classifiers = [
     "Programming Language :: Python :: 3.12",
 ]
 dependencies = [
-    "sagemaker-core>=2.3.0",
-    "sagemaker-train>=1.3.0",
+    "sagemaker-core>=2.3.1",
+    "sagemaker-train>=1.3.1",
     "boto3>=1.42.2,<2.0",
     "botocore>=1.35.75,<2.0",
     "deepdiff",
diff --git a/sagemaker-serve/tests/integ/test_huggingface_integration.py b/sagemaker-serve/tests/integ/test_huggingface_integration.py
@@ -95,10 +95,7 @@ def build_and_deploy():
     )
     
     # Build and deploy your model. Returns SageMaker Core Model and Endpoint objects
-    core_model = model_builder.build(
-        model_name=f"{MODEL_NAME_PREFIX}-{unique_id}",
-        region="us-east-1"
-    )
+    core_model = model_builder.build(model_name=f"{MODEL_NAME_PREFIX}-{unique_id}")
     logger.info(f"Model Successfully Created: {core_model.model_name}")
 
     core_endpoint = model_builder.deploy(endpoint_name=f"{ENDPOINT_NAME_PREFIX}-{unique_id}")
diff --git a/sagemaker-serve/tests/integ/test_train_inference_e2e_integration.py b/sagemaker-serve/tests/integ/test_train_inference_e2e_integration.py
@@ -33,10 +33,6 @@
 ENDPOINT_NAME_PREFIX = "train-inf-v3-test-endpoint"
 TRAINING_JOB_PREFIX = "e2e-v3-pytorch"
 
-# Configuration
-AWS_REGION = "us-west-2"
-PYTORCH_TRAINING_IMAGE = "763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training:1.13.1-cpu-py39"
-
 
 @pytest.mark.slow_test
 def test_train_inference_e2e_build_deploy_invoke_cleanup():
@@ -143,19 +139,28 @@ def create_schema_builder():
 
 def train_model():
     """Train model using ModelTrainer."""
+    from sagemaker.core import image_uris
     from sagemaker.core.helper.session_helper import Session
-    import boto3
     
-    # Create SageMaker session with AWS region
-    boto_session = boto3.Session(region_name=AWS_REGION)
-    sagemaker_session = Session(boto_session=boto_session)
+    # Get the current region from a session
+    session = Session()
+    region = session.boto_region_name
     
     training_code_dir = create_pytorch_training_code()
     unique_id = str(uuid.uuid4())[:8]
     
+    # Get training image for the current region
+    training_image = image_uris.retrieve(
+        framework="pytorch",
+        region=region,
+        version="1.13.1",
+        py_version="py39",
+        instance_type="ml.m5.xlarge",
+        image_scope="training"
+    )
+    
     model_trainer = ModelTrainer(
-        sagemaker_session=sagemaker_session,
-        training_image=PYTORCH_TRAINING_IMAGE,
+        training_image=training_image,
         source_code=SourceCode(
             source_dir=training_code_dir,
             entry_script="train.py",
@@ -173,6 +178,8 @@ def train_model():
 def build_and_deploy(model_trainer, unique_id):
     """Build and deploy model using ModelBuilder."""
     from sagemaker.serve.spec.inference_spec import InferenceSpec
+    from sagemaker.core import image_uris
+    from sagemaker.core.helper.session_helper import Session
     
     class SimpleInferenceSpec(InferenceSpec):
         def load(self, model_dir):
@@ -185,16 +192,30 @@ def invoke(self, input_object, model):
 
     schema_builder = create_schema_builder()
     
+    # Get the current region from a session
+    session = Session()
+    region = session.boto_region_name
+    
+    # Get inference image for the current region
+    inference_image = image_uris.retrieve(
+        framework="pytorch",
+        region=region,
+        version="1.13.1",
+        py_version="py39",
+        instance_type="ml.m5.xlarge",
+        image_scope="inference"
+    )
+    
     model_builder = ModelBuilder(
         model=model_trainer,
         schema_builder=schema_builder,
         model_server=ModelServer.TORCHSERVE,
         inference_spec=SimpleInferenceSpec(),
-        image_uri=PYTORCH_TRAINING_IMAGE.replace("training", "inference"),
+        image_uri=inference_image,
         dependencies={"auto": False},
     )
     
-    core_model = model_builder.build(model_name=f"{MODEL_NAME_PREFIX}-{unique_id}", region="us-west-2")
+    core_model = model_builder.build(model_name=f"{MODEL_NAME_PREFIX}-{unique_id}")
     logger.info(f"Model Successfully Created: {core_model.model_name}")
     
     core_endpoint = model_builder.deploy(
diff --git a/sagemaker-train/CHANGELOG.md b/sagemaker-train/CHANGELOG.md
@@ -1,5 +1,10 @@
 # Changelog
 
+## v1.3.1 (2026-01-12)
+
+### Bug fixes and Other Changes
+* Removing experiment_config parameter for aws_batch as it is no longer needed with the removal of Estimator
+
 ## v1.3.0 (2025-12-19)
 
 ### Features
diff --git a/sagemaker-train/VERSION b/sagemaker-train/VERSION
@@ -1 +1 @@
-1.3.0
+1.3.1
diff --git a/sagemaker-train/pyproject.toml b/sagemaker-train/pyproject.toml
@@ -33,7 +33,7 @@ classifiers = [
   "Programming Language :: Python :: 3.12",
 ]
 dependencies = [
-    "sagemaker-core>=2.3.0",
+    "sagemaker-core>=2.3.1",
     "graphene>=3,<4",
     "typing_extensions>=4.9.0",
     "tblib>=1.7.0",
diff --git a/sagemaker-train/src/sagemaker/train/aws_batch/training_queue.py b/sagemaker-train/src/sagemaker/train/aws_batch/training_queue.py
@@ -41,7 +41,6 @@ def submit(
         share_identifier: Optional[str] = None,
         timeout: Optional[Dict] = None,
         tags: Optional[Dict] = None,
-        experiment_config: Optional[Dict] = None,
     ) -> TrainingQueuedJob:
         """Submit a queued job and return a QueuedJob object.
 
@@ -54,9 +53,6 @@ def submit(
             share_identifier: Share identifier for Batch job.
             timeout: Timeout configuration for Batch job.
             tags: Tags apply to Batch job. These tags are for Batch job only.
-            experiment_config: Experiment management configuration.
-                Optionally, the dict can contain four keys:
-                'ExperimentName', 'TrialName', 'TrialComponentDisplayName' and 'RunName'.
 
         Returns: a TrainingQueuedJob object with Batch job ARN and job name.
 
@@ -71,11 +67,6 @@ def submit(
             raise ValueError(
                 "TrainingQueue requires using a ModelTrainer with Mode.SAGEMAKER_TRAINING_JOB"
             )
-        if experiment_config is not None:
-            logging.warning(
-                "ExperimentConfig is not supported for ModelTrainer. "
-                "It will be ignored when submitting the job."
-            )
         training_payload = training_job._create_training_job_args(
             input_data_config=inputs, boto3=True
         )
@@ -111,7 +102,6 @@ def map(
         share_identifier: Optional[str] = None,
         timeout: Optional[Dict] = None,
         tags: Optional[Dict] = None,
-        experiment_config: Optional[Dict] = None,
     ) -> List[TrainingQueuedJob]:
         """Submit queued jobs to the provided estimator and return a list of TrainingQueuedJob objects.
 
@@ -124,15 +114,10 @@ def map(
             share_identifier: Share identifier for the Batch jobs.
             timeout: Timeout configuration for the Batch jobs.
             tags: Tags apply to Batch job. These tags are for Batch job only.
-            experiment_config: Experiment management configuration.
-                Optionally, the dict can contain four keys:
-                'ExperimentName', 'TrialName', 'TrialComponentDisplayName' and 'RunName'.
 
         Returns: a list of TrainingQueuedJob objects with each Batch job ARN and job name.
 
         """
-        if experiment_config is None:
-            experiment_config = {}
 
         if job_names is not None:
             if len(job_names) != len(inputs):
@@ -153,7 +138,6 @@ def map(
                 share_identifier,
                 timeout,
                 tags,
-                experiment_config,
             )
             queued_batch_job_list.append(queued_batch_job)
 

Original file line number	Diff line number	Diff line change
`@@ -33,10 +33,10 @@ classifiers = [`
`33`	`33`	`"Programming Language :: Python :: 3.12",`
`34`	`34`	`]`
`35`	`35`	`dependencies = [`
`36`		`- "sagemaker-core>=2.3.0,<3.0.0",`
`37`		`- "sagemaker-train>=1.3.0,<2.0.0",`
`38`		`- "sagemaker-serve>=1.3.0,<2.0.0",`
`39`		`- "sagemaker-mlops>=1.3.0,<2.0.0",`
	`36`	`+ "sagemaker-core>=2.3.1,<3.0.0",`
	`37`	`+ "sagemaker-train>=1.3.1,<2.0.0",`
	`38`	`+ "sagemaker-serve>=1.3.1,<2.0.0",`
	`39`	`+ "sagemaker-mlops>=1.3.1,<2.0.0",`
`40`	`40`	`]`
`41`	`41`
`42`	`42`	`[project.optional-dependencies]`
Original file line number	Diff line number	Diff line change
`@@ -22,9 +22,9 @@ classifiers = [`
`22`	`22`	`"Programming Language :: Python :: 3.12",`
`23`	`23`	`]`
`24`	`24`	`dependencies = [`
`25`		`- "sagemaker-core>=2.3.0",`
`26`		`- "sagemaker-train>=1.3.0",`
`27`		`- "sagemaker-serve>=1.3.0",`
	`25`	`+ "sagemaker-core>=2.3.1",`
	`26`	`+ "sagemaker-train>=1.3.1",`
	`27`	`+ "sagemaker-serve>=1.3.1",`
`28`	`28`	`"boto3>=1.42.2,<2.0",`
`29`	`29`	`"botocore>=1.42.2,<2.0",`
`30`	`30`	`]`