sagemaker 2.239.2__tar.gz → 2.240.0__tar.gz

{sagemaker-2.239.2 → sagemaker-2.240.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sagemaker
-Version: 2.239.2
+Version: 2.240.0
 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

sagemaker-2.240.0/VERSION

@@ -0,0 +1 @@
+2.240.0

sagemaker-2.240.0/src/sagemaker/amazon/common.py

@@ -0,0 +1,25 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You
+# may not use this file except in compliance with the License. A copy of
+# the License is located at
+#
+#     http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file is
+# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.
+"""Placeholder docstring"""
+from __future__ import absolute_import
+
+# these imports ensure backward compatibility.
+from sagemaker.deserializers import RecordDeserializer  # noqa: F401 # pylint: disable=W0611
+from sagemaker.serializers import RecordSerializer  # noqa: F401 # pylint: disable=W0611
+from sagemaker.serializer_utils import (  # noqa: F401 # pylint: disable=W0611
+    read_recordio,
+    read_records,
+    write_numpy_to_dense_tensor,
+    write_spmatrix_to_sparse_tensor,
+    _write_recordio,
+)

{sagemaker-2.239.2 → sagemaker-2.240.0}/src/sagemaker/base_deserializers.py

@@ -23,7 +23,7 @@ import json
 import numpy as np
 from six import with_metaclass
 
-from sagemaker.amazon.common import read_records
+from sagemaker.serializer_utils import read_records
 from sagemaker.utils import DeferredError
 
 try:

{sagemaker-2.239.2 → sagemaker-2.240.0}/src/sagemaker/base_serializers.py

@@ -22,7 +22,7 @@ import numpy as np
 from pandas import DataFrame
 from six import with_metaclass
 
-from sagemaker.amazon.common import write_numpy_to_dense_tensor
+from sagemaker.serializer_utils import write_numpy_to_dense_tensor
 from sagemaker.utils import DeferredError
 
 try:

{sagemaker-2.239.2 → sagemaker-2.240.0}/src/sagemaker/fw_utils.py

@@ -10,30 +10,29 @@
 # distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
-"""Utility methods used by framework classes"""
+"""Utility methods used by framework classes."""
 from __future__ import absolute_import
 
 import json
 import logging
 import os
 import re
-import time
 import shutil
 import tempfile
+import time
 from collections import namedtuple
-from typing import List, Optional, Union, Dict
+from typing import Dict, List, Optional, Union
+
 from packaging import version
 
 import sagemaker.image_uris
+import sagemaker.utils
+from sagemaker.deprecations import deprecation_warn_base, renamed_kwargs, renamed_warning
 from sagemaker.instance_group import InstanceGroup
 from sagemaker.s3_utils import s3_path_join
 from sagemaker.session_settings import SessionSettings
-import sagemaker.utils
 from sagemaker.workflow import is_pipeline_variable
-
-from sagemaker.deprecations import renamed_warning, renamed_kwargs
 from sagemaker.workflow.entities import PipelineVariable
-from sagemaker.deprecations import deprecation_warn_base
 
 logger = logging.getLogger(__name__)
 
@@ -41,6 +40,7 @@ _TAR_SOURCE_FILENAME = "source.tar.gz"
 
 UploadedCode = namedtuple("UploadedCode", ["s3_prefix", "script_name"])
 """sagemaker.fw_utils.UploadedCode: An object containing the S3 prefix and script name.
+
 This is for the source code used for the entry point with an ``Estimator``. It can be
 instantiated with positional or keyword arguments.
 """
@@ -211,7 +211,7 @@ def validate_source_code_input_against_pipeline_variables(
     git_config: Optional[Dict[str, str]] = None,
     enable_network_isolation: Union[bool, PipelineVariable] = False,
 ):
-    """Validate source code input against pipeline variables
+    """Validate source code input against pipeline variables.
 
     Args:
         entry_point (str or PipelineVariable): The path to the local Python source file that
@@ -481,7 +481,7 @@ def tar_and_upload_dir(
 
 
 def _list_files_to_compress(script, directory):
-    """Placeholder docstring"""
+    """Placeholder docstring."""
     if directory is None:
         return [script]
 
@@ -585,7 +585,6 @@ def model_code_key_prefix(code_location_key_prefix, model_name, image):
     The location returned is a potential concatenation of 2 parts
         1. code_location_key_prefix if it exists
         2. model_name or a name derived from the image
-
     Args:
         code_location_key_prefix (str): the s3 key prefix from code_location
         model_name (str): the name of the model
@@ -620,8 +619,6 @@ def warn_if_parameter_server_with_multi_gpu(training_instance_type, distribution
                         "enabled": True
                     }
                 }
-
-
     """
     if training_instance_type == "local" or distribution is None:
         return
@@ -646,7 +643,7 @@ def warn_if_parameter_server_with_multi_gpu(training_instance_type, distribution
 def profiler_config_deprecation_warning(
     profiler_config, image_uri, framework_name, framework_version
 ):
-    """Put out a deprecation message for if framework profiling is specified TF >= 2.12 and PT >= 2.0"""
+    """Deprecation message if framework profiling is specified TF >= 2.12 and PT >= 2.0."""
     if profiler_config is None or profiler_config.framework_profile_params is None:
         return
 
@@ -692,6 +689,7 @@ def validate_smdistributed(
         framework_name (str): A string representing the name of framework selected.
         framework_version (str): A string representing the framework version selected.
         py_version (str): A string representing the python version selected.
+        Ex: `py38, py39, py310, py311`
         distribution (dict): A dictionary with information to enable distributed training.
             (Defaults to None if distributed training is not enabled.) For example:
 
@@ -763,7 +761,8 @@ def _validate_smdataparallel_args(
         instance_type (str): A string representing the type of training instance selected. Ex: `ml.p3.16xlarge`
         framework_name (str): A string representing the name of framework selected. Ex: `tensorflow`
         framework_version (str): A string representing the framework version selected. Ex: `2.3.1`
-        py_version (str): A string representing the python version selected. Ex: `py3`
+        py_version (str): A string representing the python version selected.
+        Ex: `py38, py39, py310, py311`
         distribution (dict): A dictionary with information to enable distributed training.
             (Defaults to None if distributed training is not enabled.) Ex:
 
@@ -847,6 +846,7 @@ def validate_distribution(
         framework_name (str): A string representing the name of framework selected.
         framework_version (str): A string representing the framework version selected.
         py_version (str): A string representing the python version selected.
+        Ex: `py38, py39, py310, py311`
         image_uri (str): A string representing a Docker image URI.
         kwargs(dict): Additional kwargs passed to this function
 
@@ -953,7 +953,7 @@ def validate_distribution(
 
 
 def validate_distribution_for_instance_type(instance_type, distribution):
-    """Check if the provided distribution strategy is supported for the instance_type
+    """Check if the provided distribution strategy is supported for the instance_type.
 
     Args:
         instance_type (str): A string representing the type of training instance selected.
@@ -1010,6 +1010,7 @@ def validate_torch_distributed_distribution(
                 }
         framework_version (str): A string representing the framework version selected.
         py_version (str): A string representing the python version selected.
+        Ex: `py38, py39, py310, py311`
         image_uri (str): A string representing a Docker image URI.
         entry_point (str or PipelineVariable): The absolute or relative path to the local Python
             source file that should be executed as the entry point to
@@ -1072,7 +1073,7 @@ def validate_torch_distributed_distribution(
 
 
 def _is_gpu_instance(instance_type):
-    """Returns bool indicating whether instance_type supports GPU
+    """Returns bool indicating whether instance_type supports GPU.
 
     Args:
         instance_type (str): Name of the instance_type to check against.
@@ -1091,7 +1092,7 @@ def _is_gpu_instance(instance_type):
 
 
 def _is_trainium_instance(instance_type):
-    """Returns bool indicating whether instance_type is a Trainium instance
+    """Returns bool indicating whether instance_type is a Trainium instance.
 
     Args:
         instance_type (str): Name of the instance_type to check against.
@@ -1107,7 +1108,7 @@ def _is_trainium_instance(instance_type):
 
 
 def python_deprecation_warning(framework, latest_supported_version):
-    """Placeholder docstring"""
+    """Placeholder docstring."""
     return PYTHON_2_DEPRECATION_WARNING.format(
         framework=framework, latest_supported_version=latest_supported_version
     )
@@ -1121,7 +1122,6 @@ def _region_supports_debugger(region_name):
 
     Returns:
         bool: Whether or not the region supports Amazon SageMaker Debugger.
-
     """
     return region_name.lower() not in DEBUGGER_UNSUPPORTED_REGIONS
 
@@ -1134,7 +1134,6 @@ def _region_supports_profiler(region_name):
 
     Returns:
         bool: Whether or not the region supports Amazon SageMaker Debugger profiling feature.
-
     """
     return region_name.lower() not in PROFILER_UNSUPPORTED_REGIONS
 
@@ -1162,7 +1161,8 @@ def validate_version_or_image_args(framework_version, py_version, image_uri):
 
     Args:
         framework_version (str): The version of the framework.
-        py_version (str): The version of Python.
+        py_version (str): A string representing the python version selected.
+        Ex: `py38, py39, py310, py311`
         image_uri (str): The URI of the image.
 
     Raises:
@@ -1194,9 +1194,8 @@ def create_image_uri(
         instance_type (str): SageMaker instance type. Used to determine device
             type (cpu/gpu/family-specific optimized).
         framework_version (str): The version of the framework.
-        py_version (str): Optional. Python version. If specified, should be one
-            of 'py2' or 'py3'. If not specified, image uri will not include a
-            python component.
+        py_version (str): Optional. Python version Ex: `py38, py39, py310, py311`.
+            If not specified, image uri will not include a python component.
         account (str): AWS account that contains the image. (default:
             '520713654638')
         accelerator_type (str): SageMaker Elastic Inference accelerator type.

{sagemaker-2.239.2 → sagemaker-2.240.0}/src/sagemaker/huggingface/estimator.py

@@ -15,17 +15,13 @@ from __future__ import absolute_import
 
 import logging
 import re
-from typing import Optional, Union, Dict
+from typing import Dict, Optional, Union
 
-from sagemaker.estimator import Framework, EstimatorBase
-from sagemaker.fw_utils import (
-    framework_name_from_image,
-    validate_distribution,
-)
+from sagemaker.estimator import EstimatorBase, Framework
+from sagemaker.fw_utils import framework_name_from_image, validate_distribution
 from sagemaker.huggingface.model import HuggingFaceModel
-from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT
-
 from sagemaker.huggingface.training_compiler.config import TrainingCompilerConfig
+from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT
 from sagemaker.workflow.entities import PipelineVariable
 
 logger = logging.getLogger("sagemaker")
@@ -66,7 +62,7 @@ class HuggingFace(Framework):
         Args:
             py_version (str): Python version you want to use for executing your model training
                 code. Defaults to ``None``. Required unless ``image_uri`` is provided.  If
-                using PyTorch, the current supported version is ``py36``. If using TensorFlow,
+                using PyTorch, the current supported version is ``py39``. If using TensorFlow,
                 the current supported version is ``py37``.
             entry_point (str or PipelineVariable): Path (absolute or relative) to the Python source
                 file which should be executed as the entry point to training.

{sagemaker-2.239.2 → sagemaker-2.240.0}/src/sagemaker/image_uri_config/djl-neuronx.json

@@ -13,12 +13,14 @@
                 "ap-southeast-1": "763104351884",
                 "ap-southeast-2": "763104351884",
                 "ap-southeast-5": "550225433462",
+                "ap-southeast-7": "590183813437",
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
                 "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
+                "mx-central-1":"637423239942",
                 "sa-east-1": "763104351884",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
@@ -37,12 +39,14 @@
                 "ap-southeast-1": "763104351884",
                 "ap-southeast-2": "763104351884",
                 "ap-southeast-5": "550225433462",
+                "ap-southeast-7": "590183813437",
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
                 "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
+                "mx-central-1":"637423239942",
                 "sa-east-1": "763104351884",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
@@ -61,12 +65,14 @@
                 "ap-southeast-1": "763104351884",
                 "ap-southeast-2": "763104351884",
                 "ap-southeast-5": "550225433462",
+                "ap-southeast-7": "590183813437",
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
                 "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
+                "mx-central-1":"637423239942",
                 "sa-east-1": "763104351884",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
@@ -85,12 +91,14 @@
                 "ap-southeast-1": "763104351884",
                 "ap-southeast-2": "763104351884",
                 "ap-southeast-5": "550225433462",
+                "ap-southeast-7": "590183813437",
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
                 "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
+                "mx-central-1":"637423239942",
                 "sa-east-1": "763104351884",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
@@ -109,12 +117,14 @@
                 "ap-southeast-1": "763104351884",
                 "ap-southeast-2": "763104351884",
                 "ap-southeast-5": "550225433462",
+                "ap-southeast-7": "590183813437",
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
                 "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
+                "mx-central-1":"637423239942",
                 "sa-east-1": "763104351884",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
@@ -133,12 +143,14 @@
                 "ap-southeast-1": "763104351884",
                 "ap-southeast-2": "763104351884",
                 "ap-southeast-5": "550225433462",
+                "ap-southeast-7": "590183813437",
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
                 "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
+                "mx-central-1":"637423239942",
                 "sa-east-1": "763104351884",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
@@ -157,12 +169,14 @@
                 "ap-southeast-1": "763104351884",
                 "ap-southeast-2": "763104351884",
                 "ap-southeast-5": "550225433462",
+                "ap-southeast-7": "590183813437",
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
                 "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
+                "mx-central-1":"637423239942",
                 "sa-east-1": "763104351884",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
@@ -181,12 +195,14 @@
                 "ap-southeast-1": "763104351884",
                 "ap-southeast-2": "763104351884",
                 "ap-southeast-5": "550225433462",
+                "ap-southeast-7": "590183813437",
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
                 "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
+                "mx-central-1":"637423239942",
                 "sa-east-1": "763104351884",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",

{sagemaker-2.239.2 → sagemaker-2.240.0}/src/sagemaker/image_uri_config/huggingface-llm-neuronx.json

@@ -4,7 +4,7 @@
             "inf2"
         ],
         "version_aliases": {
-            "0.0": "0.0.25"
+            "0.0": "0.0.27"
         },
         "versions": {
             "0.0.16": {
@@ -19,6 +19,7 @@
                     "ap-southeast-2": "763104351884",
                     "ap-southeast-4": "457447274322",
                     "ap-southeast-5": "550225433462",
+                    "ap-southeast-7": "590183813437",
                     "cn-north-1": "727897471807",
                     "cn-northwest-1": "727897471807",
                     "eu-central-1": "763104351884",
@@ -27,6 +28,7 @@
                     "eu-west-1": "763104351884",
                     "eu-west-3": "763104351884",
                     "il-central-1": "780543022126",
+                    "mx-central-1":"637423239942",
                     "sa-east-1": "763104351884",
                     "us-east-1": "763104351884",
                     "us-east-2": "763104351884",
@@ -53,6 +55,7 @@
                     "ap-southeast-2": "763104351884",
                     "ap-southeast-4": "457447274322",
                     "ap-southeast-5": "550225433462",
+                    "ap-southeast-7": "590183813437",
                     "cn-north-1": "727897471807",
                     "cn-northwest-1": "727897471807",
                     "eu-central-1": "763104351884",
@@ -61,6 +64,7 @@
                     "eu-west-1": "763104351884",
                     "eu-west-3": "763104351884",
                     "il-central-1": "780543022126",
+                    "mx-central-1":"637423239942",
                     "sa-east-1": "763104351884",
                     "us-east-1": "763104351884",
                     "us-east-2": "763104351884",
@@ -87,6 +91,7 @@
                     "ap-southeast-2": "763104351884",
                     "ap-southeast-4": "457447274322",
                     "ap-southeast-5": "550225433462",
+                    "ap-southeast-7": "590183813437",
                     "cn-north-1": "727897471807",
                     "cn-northwest-1": "727897471807",
                     "eu-central-1": "763104351884",
@@ -95,6 +100,7 @@
                     "eu-west-1": "763104351884",
                     "eu-west-3": "763104351884",
                     "il-central-1": "780543022126",
+                    "mx-central-1":"637423239942",
                     "sa-east-1": "763104351884",
                     "us-east-1": "763104351884",
                     "us-east-2": "763104351884",
@@ -121,6 +127,7 @@
                     "ap-southeast-2": "763104351884",
                     "ap-southeast-4": "457447274322",
                     "ap-southeast-5": "550225433462",
+                    "ap-southeast-7": "590183813437",
                     "cn-north-1": "727897471807",
                     "cn-northwest-1": "727897471807",
                     "eu-central-1": "763104351884",
@@ -129,6 +136,7 @@
                     "eu-west-1": "763104351884",
                     "eu-west-3": "763104351884",
                     "il-central-1": "780543022126",
+                    "mx-central-1":"637423239942",
                     "sa-east-1": "763104351884",
                     "us-east-1": "763104351884",
                     "us-east-2": "763104351884",
@@ -155,6 +163,7 @@
                     "ap-southeast-2": "763104351884",
                     "ap-southeast-4": "457447274322",
                     "ap-southeast-5": "550225433462",
+                    "ap-southeast-7": "590183813437",
                     "cn-north-1": "727897471807",
                     "cn-northwest-1": "727897471807",
                     "eu-central-1": "763104351884",
@@ -163,6 +172,7 @@
                     "eu-west-1": "763104351884",
                     "eu-west-3": "763104351884",
                     "il-central-1": "780543022126",
+                    "mx-central-1":"637423239942",
                     "sa-east-1": "763104351884",
                     "us-east-1": "763104351884",
                     "us-east-2": "763104351884",
@@ -189,6 +199,7 @@
                     "ap-southeast-2": "763104351884",
                     "ap-southeast-4": "457447274322",
                     "ap-southeast-5": "550225433462",
+                    "ap-southeast-7": "590183813437",
                     "cn-north-1": "727897471807",
                     "cn-northwest-1": "727897471807",
                     "eu-central-1": "763104351884",
@@ -197,6 +208,7 @@
                     "eu-west-1": "763104351884",
                     "eu-west-3": "763104351884",
                     "il-central-1": "780543022126",
+                    "mx-central-1":"637423239942",
                     "sa-east-1": "763104351884",
                     "us-east-1": "763104351884",
                     "us-east-2": "763104351884",
@@ -223,6 +235,7 @@
                     "ap-southeast-2": "763104351884",
                     "ap-southeast-4": "457447274322",
                     "ap-southeast-5": "550225433462",
+                    "ap-southeast-7": "590183813437",
                     "cn-north-1": "727897471807",
                     "cn-northwest-1": "727897471807",
                     "eu-central-1": "763104351884",
@@ -231,6 +244,7 @@
                     "eu-west-1": "763104351884",
                     "eu-west-3": "763104351884",
                     "il-central-1": "780543022126",
+                    "mx-central-1":"637423239942",
                     "sa-east-1": "763104351884",
                     "us-east-1": "763104351884",
                     "us-east-2": "763104351884",
@@ -255,6 +269,7 @@
                     "ap-southeast-2": "763104351884",
                     "ap-southeast-4": "457447274322",
                     "ap-southeast-5": "550225433462",
+                    "ap-southeast-7": "590183813437",
                     "cn-north-1": "727897471807",
                     "cn-northwest-1": "727897471807",
                     "eu-central-1": "763104351884",
@@ -263,6 +278,7 @@
                     "eu-west-1": "763104351884",
                     "eu-west-3": "763104351884",
                     "il-central-1": "780543022126",
+                    "mx-central-1":"637423239942",
                     "sa-east-1": "763104351884",
                     "us-east-1": "763104351884",
                     "us-east-2": "763104351884",
@@ -289,6 +305,7 @@
                     "ap-southeast-2": "763104351884",
                     "ap-southeast-4": "457447274322",
                     "ap-southeast-5": "550225433462",
+                    "ap-southeast-7": "590183813437",
                     "cn-north-1": "727897471807",
                     "cn-northwest-1": "727897471807",
                     "eu-central-1": "763104351884",
@@ -297,6 +314,7 @@
                     "eu-west-1": "763104351884",
                     "eu-west-3": "763104351884",
                     "il-central-1": "780543022126",
+                    "mx-central-1":"637423239942",
                     "sa-east-1": "763104351884",
                     "us-east-1": "763104351884",
                     "us-east-2": "763104351884",
@@ -323,6 +341,7 @@
                     "ap-southeast-2": "763104351884",
                     "ap-southeast-4": "457447274322",
                     "ap-southeast-5": "550225433462",
+                    "ap-southeast-7": "590183813437",
                     "cn-north-1": "727897471807",
                     "cn-northwest-1": "727897471807",
                     "eu-central-1": "763104351884",
@@ -331,6 +350,7 @@
                     "eu-west-1": "763104351884",
                     "eu-west-3": "763104351884",
                     "il-central-1": "780543022126",
+                    "mx-central-1":"637423239942",
                     "sa-east-1": "763104351884",
                     "us-east-1": "763104351884",
                     "us-east-2": "763104351884",
@@ -344,6 +364,40 @@
                 "container_version": {
                     "inf2": "ubuntu22.04"
                 }
+            },
+            "0.0.27": {
+                "py_versions": [
+                    "py310"
+                ],
+                "registries": {
+                    "ap-northeast-1": "763104351884",
+                    "ap-south-1": "763104351884",
+                    "ap-south-2": "772153158452",
+                    "ap-southeast-1": "763104351884",
+                    "ap-southeast-2": "763104351884",
+                    "ap-southeast-4": "457447274322",
+                    "ap-southeast-5": "550225433462",
+                    "cn-north-1": "727897471807",
+                    "cn-northwest-1": "727897471807",
+                    "eu-central-1": "763104351884",
+                    "eu-central-2": "380420809688",
+                    "eu-south-2": "503227376785",
+                    "eu-west-1": "763104351884",
+                    "eu-west-3": "763104351884",
+                    "il-central-1": "780543022126",
+                    "sa-east-1": "763104351884",
+                    "us-east-1": "763104351884",
+                    "us-east-2": "763104351884",
+                    "us-gov-east-1": "446045086412",
+                    "us-gov-west-1": "442386744353",
+                    "us-west-2": "763104351884",
+                    "ca-west-1": "204538143572"
+                },
+                "tag_prefix": "2.1.2-optimum0.0.27",
+                "repository": "huggingface-pytorch-tgi-inference",
+                "container_version": {
+                    "inf2": "ubuntu22.04"
+                }
             }
         }
     }

{sagemaker-2.239.2 → sagemaker-2.240.0}/src/sagemaker/image_uri_config/huggingface-neuron.json

@@ -24,12 +24,14 @@
                         "ap-southeast-2": "763104351884",
                         "ap-southeast-4": "457447274322",
                         "ap-southeast-5": "550225433462",
+                        "ap-southeast-7": "590183813437",
                         "eu-central-1": "763104351884",
                         "eu-central-2": "380420809688",
                         "eu-south-2": "503227376785",
                         "eu-west-1": "763104351884",
                         "eu-west-3": "763104351884",
                         "il-central-1": "780543022126",
+                        "mx-central-1":"637423239942",
                         "sa-east-1": "763104351884",
                         "us-east-1": "763104351884",
                         "us-east-2": "763104351884",