sagemaker-core 2.25.0__tar.gz → 2.26.0__tar.gz

{sagemaker_core-2.25.0/src/sagemaker_core.egg-info → sagemaker_core-2.26.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 2.25.0
+Version: 2.26.0
 Summary: An python package for sagemaker core functionalities
 Author-email: AWS <sagemaker-interests@amazon.com>
 Project-URL: Homepage, https://github.com/aws/sagemaker-python-sdk

sagemaker_core-2.26.0/VERSION

@@ -0,0 +1 @@
+2.26.0

{sagemaker_core-2.25.0 → sagemaker_core-2.26.0}/src/sagemaker/core/image_uri_config/djl-neuronx.json

@@ -23,6 +23,7 @@
                 "eu-west-3": "763104351884",
                 "mx-central-1": "637423239942",
                 "sa-east-1": "763104351884",
+                "sa-west-1": "434457397001",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
                 "us-gov-east-1": "446045086412",
@@ -59,6 +60,7 @@
                 "eu-west-3": "763104351884",
                 "mx-central-1": "637423239942",
                 "sa-east-1": "763104351884",
+                "sa-west-1": "434457397001",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
                 "us-gov-east-1": "446045086412",
@@ -91,6 +93,7 @@
                 "eu-west-3": "763104351884",
                 "mx-central-1": "637423239942",
                 "sa-east-1": "763104351884",
+                "sa-west-1": "434457397001",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
                 "us-gov-east-1": "446045086412",
@@ -123,6 +126,7 @@
                 "eu-west-3": "763104351884",
                 "mx-central-1": "637423239942",
                 "sa-east-1": "763104351884",
+                "sa-west-1": "434457397001",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
                 "us-gov-east-1": "446045086412",
@@ -155,6 +159,7 @@
                 "eu-west-3": "763104351884",
                 "mx-central-1": "637423239942",
                 "sa-east-1": "763104351884",
+                "sa-west-1": "434457397001",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
                 "us-gov-east-1": "446045086412",
@@ -185,6 +190,7 @@
                 "eu-west-3": "763104351884",
                 "mx-central-1": "637423239942",
                 "sa-east-1": "763104351884",
+                "sa-west-1": "434457397001",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
                 "us-gov-east-1": "446045086412",
@@ -215,6 +221,7 @@
                 "eu-west-3": "763104351884",
                 "mx-central-1": "637423239942",
                 "sa-east-1": "763104351884",
+                "sa-west-1": "434457397001",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
                 "us-gov-east-1": "446045086412",
@@ -245,6 +252,7 @@
                 "eu-west-3": "763104351884",
                 "mx-central-1": "637423239942",
                 "sa-east-1": "763104351884",
+                "sa-west-1": "434457397001",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
                 "us-gov-east-1": "446045086412",

{sagemaker_core-2.25.0 → sagemaker_core-2.26.0}/src/sagemaker/core/jumpstart/region_config.json

@@ -139,6 +139,10 @@
         "content_bucket": "jumpstart-cache-prod-me-south-1",
         "gated_content_bucket": "jumpstart-private-cache-prod-me-south-1"
     },
+    "me-west-1": {
+        "content_bucket": "jumpstart-cache-prod-me-west-1",
+        "gated_content_bucket": "jumpstart-private-cache-prod-me-west-1"
+    },
     "mx-central-1": {
         "content_bucket": "jumpstart-cache-prod-mx-central-1",
         "gated_content_bucket": "jumpstart-private-cache-prod-mx-central-1"
@@ -148,6 +152,10 @@
         "gated_content_bucket": "jumpstart-private-cache-prod-sa-east-1",
         "neo_content_bucket": "sagemaker-sd-models-prod-sa-east-1"
     },
+    "sa-west-1": {
+        "content_bucket": "jumpstart-cache-prod-sa-west-1",
+        "gated_content_bucket": "jumpstart-private-cache-prod-sa-west-1"
+    },
     "us-east-1": {
         "content_bucket": "jumpstart-cache-prod-us-east-1",
         "gated_content_bucket": "jumpstart-private-cache-prod-us-east-1",

{sagemaker_core-2.25.0 → sagemaker_core-2.26.0}/src/sagemaker/core/model_registry.py

@@ -15,12 +15,33 @@ from sagemaker.core.config import (
     MODEL_PACKAGE_VALIDATION_PROFILES_PATH,
 )
 from sagemaker.core.resources import ModelPackageModelCard
+from sagemaker.core.utils.utils import serialize
 from botocore.exceptions import ClientError
 import logging
 
 logger = LOGGER = logging.getLogger("sagemaker")
 
 
+def _to_request_dict(value):
+    """Convert a model registry argument into its CreateModelPackage request shape.
+
+    Supports both the modern ``sagemaker.core.shapes`` pydantic shapes, which are
+    converted with the core serializer, and the older compatibility classes in
+    ``sagemaker.core.model_metrics`` and friends, which implement their own
+    ``_to_request_dict()``. Plain dicts are assumed to already be in request form
+    and are passed through the serializer unchanged.
+
+    Args:
+        value: A core shape, a compatibility class instance, or a request dict.
+
+    Returns:
+        dict: The serialized request shape.
+    """
+    if hasattr(value, "_to_request_dict"):
+        return value._to_request_dict()
+    return serialize(value)
+
+
 def get_model_package_args(
     content_types=None,
     response_types=None,
@@ -76,11 +97,11 @@ def get_model_package_args(
     if model_package_group_name is not None:
         model_package_args["model_package_group_name"] = model_package_group_name
     if model_metrics is not None:
-        model_package_args["model_metrics"] = model_metrics._to_request_dict()
+        model_package_args["model_metrics"] = _to_request_dict(model_metrics)
     if drift_check_baselines is not None:
-        model_package_args["drift_check_baselines"] = drift_check_baselines._to_request_dict()
+        model_package_args["drift_check_baselines"] = _to_request_dict(drift_check_baselines)
     if metadata_properties is not None:
-        model_package_args["metadata_properties"] = metadata_properties._to_request_dict()
+        model_package_args["metadata_properties"] = _to_request_dict(metadata_properties)
     if approval_status is not None:
         model_package_args["approval_status"] = approval_status
     if description is not None:
@@ -102,7 +123,7 @@ def get_model_package_args(
     if source_uri is not None:
         model_package_args["source_uri"] = source_uri
     if model_life_cycle is not None:
-        model_package_args["model_life_cycle"] = model_life_cycle._to_request_dict()
+        model_package_args["model_life_cycle"] = _to_request_dict(model_life_cycle)
     if model_card is not None:
         original_req = {}
         if isinstance(model_card, ModelPackageModelCard):

{sagemaker_core-2.25.0 → sagemaker_core-2.26.0/src/sagemaker_core.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 2.25.0
+Version: 2.26.0
 Summary: An python package for sagemaker core functionalities
 Author-email: AWS <sagemaker-interests@amazon.com>
 Project-URL: Homepage, https://github.com/aws/sagemaker-python-sdk

sagemaker_core-2.25.0/VERSION

@@ -1 +0,0 @@
-2.25.0