sagemaker 2.243.0__tar.gz → 2.243.2__tar.gz

{sagemaker-2.243.0 → sagemaker-2.243.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sagemaker
-Version: 2.243.0
+Version: 2.243.2
 Summary: Open source library for training and deploying models on Amazon SageMaker.
 Project-URL: Homepage, https://github.com/aws/sagemaker-python-sdk
 Author: Amazon Web Services
@@ -97,6 +97,7 @@ Requires-Dist: nest-asyncio; extra == 'test'
 Requires-Dist: numpy>=1.24.0; extra == 'test'
 Requires-Dist: onnx==1.17.0; extra == 'test'
 Requires-Dist: opentelemetry-proto==1.27.0; extra == 'test'
+Requires-Dist: orderly-set<5.4.0; extra == 'test'
 Requires-Dist: pandas==1.4.4; extra == 'test'
 Requires-Dist: pillow<=11,>=10.0.1; extra == 'test'
 Requires-Dist: protobuf==4.25.5; extra == 'test'

sagemaker-2.243.2/VERSION

@@ -0,0 +1 @@
+2.243.2

{sagemaker-2.243.0 → sagemaker-2.243.2}/requirements/extras/test_requirements.txt

@@ -50,3 +50,4 @@ fastapi==0.115.4
 nest-asyncio
 sagemaker-mlflow>=0.1.0
 deepdiff>=8.0.0
+orderly-set<5.4.0

{sagemaker-2.243.0 → sagemaker-2.243.2}/src/sagemaker/image_uri_config/instance_gpu_info.json

@@ -23,7 +23,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "ap-east-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -49,7 +49,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "ap-northeast-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -75,7 +75,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "ap-northeast-2": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -101,7 +101,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "ap-northeast-3": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -127,7 +127,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "ap-south-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -153,7 +153,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "ap-southeast-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -179,7 +179,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "ap-southeast-2": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -205,7 +205,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "ap-southeast-3": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -231,7 +231,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "ca-central-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -257,7 +257,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "cn-north-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -283,7 +283,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "cn-northwest-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -309,7 +309,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "eu-central-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -335,7 +335,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "eu-central-2": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -361,7 +361,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "eu-north-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -387,7 +387,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "eu-south-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -413,7 +413,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "eu-south-2": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -439,7 +439,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "eu-west-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -465,7 +465,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "eu-west-2": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -491,7 +491,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "eu-west-3": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -517,7 +517,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "il-central-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -543,7 +543,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "me-central-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -569,7 +569,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "me-south-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -595,7 +595,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "sa-east-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -621,7 +621,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "us-east-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -647,7 +647,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "us-east-2": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -673,7 +673,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "us-gov-east-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -699,7 +699,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "us-gov-west-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -725,7 +725,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "us-west-1": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -751,7 +751,7 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     },
     "us-west-2": {
         "ml.p5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 655360},
@@ -777,6 +777,6 @@
         "ml.g5.16xlarge": {"Count": 1, "TotalGpuMemoryInMiB": 24576},
         "ml.g5.12xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
         "ml.g5.24xlarge": {"Count": 4, "TotalGpuMemoryInMiB": 98304},
-        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 196608}
+        "ml.g5.48xlarge": {"Count": 8, "TotalGpuMemoryInMiB": 183104}
     }
 }

{sagemaker-2.243.0 → sagemaker-2.243.2}/src/sagemaker/jumpstart/accessors.py

@@ -25,6 +25,7 @@ from sagemaker.jumpstart import cache
 from sagemaker.jumpstart.hub.utils import (
     construct_hub_model_arn_from_inputs,
     construct_hub_model_reference_arn_from_inputs,
+    generate_hub_arn_for_init_kwargs,
 )
 from sagemaker.jumpstart.constants import JUMPSTART_DEFAULT_REGION_NAME
 from sagemaker.session import Session
@@ -291,6 +292,10 @@ class JumpStartModelsAccessor(object):
         # Users only input model id, not contentType, so first try to describe with ModelReference, then with Model
         if hub_arn:
             try:
+                hub_arn = generate_hub_arn_for_init_kwargs(
+                    hub_name=hub_arn, region=region, session=sagemaker_session
+                )
+
                 hub_model_arn = construct_hub_model_reference_arn_from_inputs(
                     hub_arn=hub_arn, model_name=model_id, version=version
                 )

{sagemaker-2.243.0 → sagemaker-2.243.2}/src/sagemaker/jumpstart/cache.py

@@ -552,6 +552,12 @@ class JumpStartModelsCache:
                 )
             return version_str if version_str in available_versions else None
 
+        if version_str[-1] == "*":
+            # major or minor version is pinned, e.g 1.* or 1.0.*
+            return utils.get_latest_version(
+                [version for version in available_versions if version.startswith(version_str[:-1])]
+            )
+
         try:
             spec = SpecifierSet(f"=={version_str}")
         except InvalidSpecifier:

{sagemaker-2.243.0 → sagemaker-2.243.2}/src/sagemaker/jumpstart/estimator.py

@@ -41,7 +41,7 @@ from sagemaker.jumpstart.utils import (
     validate_model_id_and_get_type,
     resolve_model_sagemaker_config_field,
     verify_model_region_and_return_specs,
-    remove_env_var_from_estimator_kwargs_if_accept_eula_present,
+    remove_env_var_from_estimator_kwargs_if_model_access_config_present,
     get_model_access_config,
     get_hub_access_config,
 )
@@ -616,6 +616,7 @@ class JumpStartEstimator(Estimator):
         self.tolerate_vulnerable_model = estimator_init_kwargs.tolerate_vulnerable_model
         self.instance_count = estimator_init_kwargs.instance_count
         self.region = estimator_init_kwargs.region
+        self.environment = estimator_init_kwargs.environment
         self.orig_predictor_cls = None
         self.role = estimator_init_kwargs.role
         self.sagemaker_session = estimator_init_kwargs.sagemaker_session
@@ -693,7 +694,7 @@ class JumpStartEstimator(Estimator):
                 accept the end-user license agreement (EULA) that some
                 models require. (Default: None).
         """
-        self.model_access_config = get_model_access_config(accept_eula)
+        self.model_access_config = get_model_access_config(accept_eula, self.environment)
         self.hub_access_config = get_hub_access_config(
             hub_content_arn=self.init_kwargs.get("model_reference_arn", None)
         )
@@ -713,7 +714,9 @@ class JumpStartEstimator(Estimator):
             config_name=self.config_name,
             hub_access_config=self.hub_access_config,
         )
-        remove_env_var_from_estimator_kwargs_if_accept_eula_present(self.init_kwargs, accept_eula)
+        remove_env_var_from_estimator_kwargs_if_model_access_config_present(
+            self.init_kwargs, self.model_access_config
+        )
 
         return super(JumpStartEstimator, self).fit(**estimator_fit_kwargs.to_kwargs_dict())
 

{sagemaker-2.243.0 → sagemaker-2.243.2}/src/sagemaker/jumpstart/utils.py

@@ -1632,17 +1632,29 @@ def get_draft_model_content_bucket(provider: Dict, region: str) -> str:
     return neo_bucket
 
 
-def remove_env_var_from_estimator_kwargs_if_accept_eula_present(
-    init_kwargs: dict, accept_eula: Optional[bool]
+def remove_env_var_from_estimator_kwargs_if_model_access_config_present(
+    init_kwargs: dict, model_access_config: Optional[dict]
 ):
-    """Remove env vars if access configs are used
+    """Remove env vars if ModelAccessConfig is used
 
     Args:
         init_kwargs (dict): Dictionary of kwargs when Estimator is instantiated.
         accept_eula (Optional[bool]): Whether or not the EULA was accepted, optionally passed in to Estimator.fit().
     """
-    if accept_eula is not None and init_kwargs["environment"]:
-        del init_kwargs["environment"][constants.SAGEMAKER_GATED_MODEL_S3_URI_TRAINING_ENV_VAR_KEY]
+    if (
+        model_access_config is not None
+        and init_kwargs.get("environment") is not None
+        and init_kwargs.get("model_uri") is not None
+    ):
+        if (
+            constants.SAGEMAKER_GATED_MODEL_S3_URI_TRAINING_ENV_VAR_KEY
+            in init_kwargs["environment"]
+        ):
+            del init_kwargs["environment"][
+                constants.SAGEMAKER_GATED_MODEL_S3_URI_TRAINING_ENV_VAR_KEY
+            ]
+        if "accept_eula" in init_kwargs["environment"]:
+            del init_kwargs["environment"]["accept_eula"]
 
 
 def get_hub_access_config(hub_content_arn: Optional[str]):
@@ -1659,16 +1671,24 @@ def get_hub_access_config(hub_content_arn: Optional[str]):
     return hub_access_config
 
 
-def get_model_access_config(accept_eula: Optional[bool]):
+def get_model_access_config(accept_eula: Optional[bool], environment: Optional[dict]):
     """Get access configs
 
     Args:
         accept_eula (Optional[bool]): Whether or not the EULA was accepted, optionally passed in to Estimator.fit().
     """
+    env_var_eula = environment.get("accept_eula") if environment else None
+    if env_var_eula is not None and accept_eula is not None:
+        raise ValueError(
+            "Cannot pass in both accept_eula and environment variables. "
+            "Please remove the environment variable and pass in the accept_eula parameter."
+        )
+
+    model_access_config = None
+    if env_var_eula is not None:
+        model_access_config = {"AcceptEula": env_var_eula == "true"}
     if accept_eula is not None:
         model_access_config = {"AcceptEula": accept_eula}
-    else:
-        model_access_config = None
 
     return model_access_config
 

{sagemaker-2.243.0 → sagemaker-2.243.2}/src/sagemaker/processing.py

@@ -17,7 +17,7 @@ data pre-processing, post-processing, feature engineering, data validation, and
 and interpretation on Amazon SageMaker.
 """
 from __future__ import absolute_import
-
+import json
 import logging
 import os
 import pathlib
@@ -315,6 +315,16 @@ class Processor(object):
                 + "rather than a pipeline variable"
             )
 
+        if arguments is not None:
+            processed_arguments = []
+            for arg in arguments:
+                if isinstance(arg, PipelineVariable):
+                    processed_value = json.dumps(arg.expr)
+                    processed_arguments.append(processed_value)
+                else:
+                    processed_arguments.append(arg)
+            arguments = processed_arguments
+
         self._current_job_name = self._generate_current_job_name(job_name=job_name)
 
         inputs_with_code = self._include_code_in_inputs(inputs, code, kms_key)

{sagemaker-2.243.0 → sagemaker-2.243.2}/src/sagemaker/serve/utils/telemetry_logger.py

@@ -19,7 +19,7 @@ import requests
 
 from sagemaker import Session, exceptions
 from sagemaker.serve.mode.function_pointers import Mode
-from sagemaker.serve.model_format.mlflow.constants import MLFLOW_MODEL_PATH
+from sagemaker.serve.model_format.mlflow.constants import MLFLOW_MODEL_PATH, MLFLOW_TRACKING_ARN
 from sagemaker.serve.utils.exceptions import ModelBuilderException
 from sagemaker.serve.utils.lineage_constants import (
     MLFLOW_LOCAL_PATH,
@@ -144,6 +144,9 @@ def _capture_telemetry(func_name: str):
                 mlflow_model_path = self.model_metadata[MLFLOW_MODEL_PATH]
                 mlflow_model_path_type = _get_mlflow_model_path_type(mlflow_model_path)
                 extra += f"&x-mlflowModelPathType={MLFLOW_MODEL_PATH_CODE[mlflow_model_path_type]}"
+                mlflow_model_tracking_server_arn = self.model_metadata.get(MLFLOW_TRACKING_ARN)
+                if mlflow_model_tracking_server_arn is not None:
+                    extra += f"&x-mlflowTrackingServerArn={mlflow_model_tracking_server_arn}"
 
             if getattr(self, "model_hub", False):
                 extra += f"&x-modelHub={MODEL_HUB_TO_CODE[str(self.model_hub)]}"

{sagemaker-2.243.0 → sagemaker-2.243.2}/src/sagemaker/workflow/notebook_job_step.py

@@ -13,49 +13,33 @@
 """The notebook job step definitions for workflow."""
 from __future__ import absolute_import
 
+import os
 import re
 import shutil
-import os
+from typing import Dict, List, Optional, Union
 
-from typing import (
-    List,
-    Optional,
-    Union,
-    Dict,
+from sagemaker import vpc_utils
+from sagemaker.config.config_schema import (
+    NOTEBOOK_JOB_ROLE_ARN,
+    NOTEBOOK_JOB_S3_KMS_KEY_ID,
+    NOTEBOOK_JOB_S3_ROOT_URI,
+    NOTEBOOK_JOB_VOLUME_KMS_KEY_ID,
+    NOTEBOOK_JOB_VPC_CONFIG_SECURITY_GROUP_IDS,
+    NOTEBOOK_JOB_VPC_CONFIG_SUBNETS,
 )
-
+from sagemaker.s3 import S3Uploader
+from sagemaker.s3_utils import s3_path_join
+from sagemaker.session import get_execution_role
+from sagemaker.utils import Tags, _tmpdir, format_tags, name_from_base, resolve_value_from_config
+from sagemaker.workflow.entities import PipelineVariable, RequestType
 from sagemaker.workflow.execution_variables import ExecutionVariables
 from sagemaker.workflow.functions import Join
 from sagemaker.workflow.properties import Properties
 from sagemaker.workflow.retry import RetryPolicy
-from sagemaker.workflow.steps import (
-    Step,
-    ConfigurableRetryStep,
-    StepTypeEnum,
-)
 from sagemaker.workflow.step_collections import StepCollection
 from sagemaker.workflow.step_outputs import StepOutput
-
-from sagemaker.workflow.entities import (
-    RequestType,
-    PipelineVariable,
-)
+from sagemaker.workflow.steps import ConfigurableRetryStep, Step, StepTypeEnum
 from sagemaker.workflow.utilities import _collect_parameters, load_step_compilation_context
-from sagemaker.session import get_execution_role
-
-from sagemaker.s3_utils import s3_path_join
-from sagemaker.s3 import S3Uploader
-from sagemaker.utils import _tmpdir, name_from_base, resolve_value_from_config, format_tags, Tags
-from sagemaker import vpc_utils
-
-from sagemaker.config.config_schema import (
-    NOTEBOOK_JOB_ROLE_ARN,
-    NOTEBOOK_JOB_S3_ROOT_URI,
-    NOTEBOOK_JOB_S3_KMS_KEY_ID,
-    NOTEBOOK_JOB_VOLUME_KMS_KEY_ID,
-    NOTEBOOK_JOB_VPC_CONFIG_SUBNETS,
-    NOTEBOOK_JOB_VPC_CONFIG_SECURITY_GROUP_IDS,
-)
 
 
 # disable E1101 as collect_parameters decorator sets the attributes
@@ -374,7 +358,7 @@ class NotebookJobStep(ConfigurableRetryStep):
         execution mechanism.
         """
 
-        job_envs = self.environment_variables if self.environment_variables else {}
+        job_envs = dict(self.environment_variables or {})
         system_envs = {
             "AWS_DEFAULT_REGION": self._region_from_session,
             "SM_JOB_DEF_VERSION": "1.0",

{sagemaker-2.243.0 → sagemaker-2.243.2}/src/sagemaker/workflow/steps.py

@@ -18,7 +18,6 @@ import warnings
 
 from enum import Enum
 from typing import Dict, List, Set, Union, Optional, Any, TYPE_CHECKING
-from urllib.parse import urlparse
 
 import attr
 
@@ -465,6 +464,7 @@ class TrainingStep(ConfigurableRetryStep):
         self.step_args = step_args
         self.estimator = estimator
         self.inputs = inputs
+        self.job_name = None
 
         self._properties = Properties(
             step_name=name, step=self, shape_name="DescribeTrainingJobResponse"
@@ -493,19 +493,6 @@ class TrainingStep(ConfigurableRetryStep):
                 DeprecationWarning,
             )
 
-        self.job_name = None
-        if estimator and (estimator.source_dir or estimator.entry_point):
-            # By default, `Estimator` will upload the local code to an S3 path
-            # containing a timestamp. This causes cache misses whenever a
-            # pipeline is updated, even if the underlying script hasn't changed.
-            # To avoid this, hash the contents of the training script and include it
-            # in the `job_name` passed to the `Estimator`, which will be used
-            # instead of the timestamped path.
-            if not is_pipeline_variable(estimator.source_dir) and not is_pipeline_variable(
-                estimator.entry_point
-            ):
-                self.job_name = self._generate_code_upload_path()
-
     @property
     def arguments(self) -> RequestType:
         """The arguments dictionary that is used to call `create_training_job`.
@@ -554,26 +541,6 @@ class TrainingStep(ConfigurableRetryStep):
 
         return request_dict
 
-    def _generate_code_upload_path(self) -> str or None:
-        """Generate an upload path for local training scripts based on their content."""
-        from sagemaker.workflow.utilities import hash_files_or_dirs
-
-        if self.estimator.source_dir:
-            source_dir_url = urlparse(self.estimator.source_dir)
-            if source_dir_url.scheme == "" or source_dir_url.scheme == "file":
-                code_hash = hash_files_or_dirs(
-                    [self.estimator.source_dir] + self.estimator.dependencies
-                )
-                return f"{self.name}-{code_hash}"[:1024]
-        elif self.estimator.entry_point:
-            entry_point_url = urlparse(self.estimator.entry_point)
-            if entry_point_url.scheme == "" or entry_point_url.scheme == "file":
-                code_hash = hash_files_or_dirs(
-                    [self.estimator.entry_point] + self.estimator.dependencies
-                )
-                return f"{self.name}-{code_hash}"[:1024]
-        return None
-
 
 class CreateModelStep(ConfigurableRetryStep):
     """`CreateModelStep` for SageMaker Pipelines Workflows."""
@@ -895,16 +862,6 @@ class ProcessingStep(ConfigurableRetryStep):
                         "code argument has to be a valid S3 URI or local file path "
                         + "rather than a pipeline variable"
                     )
-                code_url = urlparse(code)
-                if code_url.scheme == "" or code_url.scheme == "file":
-                    # By default, `Processor` will upload the local code to an S3 path
-                    # containing a timestamp. This causes cache misses whenever a
-                    # pipeline is updated, even if the underlying script hasn't changed.
-                    # To avoid this, hash the contents of the script and include it
-                    # in the `job_name` passed to the `Processor`, which will be used
-                    # instead of the timestamped path.
-                    self.job_name = self._generate_code_upload_path()
-
             warnings.warn(
                 (
                     'We are deprecating the instantiation of ProcessingStep using "processor".'

sagemaker-2.243.0/VERSION

@@ -1 +0,0 @@
-2.243.0