sagemaker 2.232.1__tar.gz → 2.232.3__tar.gz

{sagemaker-2.232.1 → sagemaker-2.232.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sagemaker
-Version: 2.232.1
+Version: 2.232.3
 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
@@ -34,6 +34,7 @@ Requires-Dist: psutil
 Requires-Dist: pyyaml~=6.0
 Requires-Dist: requests
 Requires-Dist: sagemaker-core<2.0.0,>=1.0.0
+Requires-Dist: sagemaker-mlflow
 Requires-Dist: schema
 Requires-Dist: smdebug-rulesconfig==1.0.1
 Requires-Dist: tblib<4,>=1.7.0
@@ -44,7 +45,7 @@ Requires-Dist: accelerate<=0.27.0,>=0.24.1; extra == 'all'
 Requires-Dist: docker<8.0.0,>=5.0.2; extra == 'all'
 Requires-Dist: fastapi>=0.111.0; extra == 'all'
 Requires-Dist: nest-asyncio; extra == 'all'
-Requires-Dist: pyspark==3.3.1; extra == 'all'
+Requires-Dist: pyspark==3.3.2; extra == 'all'
 Requires-Dist: pyyaml<7,>=5.4.1; extra == 'all'
 Requires-Dist: sagemaker-feature-store-pyspark-3-3; extra == 'all'
 Requires-Dist: sagemaker-schema-inference-artifacts>=0.0.5; extra == 'all'
@@ -52,7 +53,7 @@ Requires-Dist: scipy==1.10.1; extra == 'all'
 Requires-Dist: urllib3<3.0.0,>=1.26.8; extra == 'all'
 Requires-Dist: uvicorn>=0.30.1; extra == 'all'
 Provides-Extra: feature-processor
-Requires-Dist: pyspark==3.3.1; extra == 'feature-processor'
+Requires-Dist: pyspark==3.3.2; extra == 'feature-processor'
 Requires-Dist: sagemaker-feature-store-pyspark-3-3; extra == 'feature-processor'
 Provides-Extra: huggingface
 Requires-Dist: accelerate<=0.27.0,>=0.24.1; extra == 'huggingface'
@@ -74,7 +75,7 @@ Requires-Dist: attrs<24,>=23.1.0; extra == 'test'
 Requires-Dist: awslogs==0.14.0; extra == 'test'
 Requires-Dist: black==24.3.0; extra == 'test'
 Requires-Dist: build[virtualenv]==1.2.1; extra == 'test'
-Requires-Dist: cloudpickle==2.2.1; extra == 'test'
+Requires-Dist: cloudpickle>=2.2.1; extra == 'test'
 Requires-Dist: contextlib2==21.6.0; extra == 'test'
 Requires-Dist: coverage<6.2,>=5.2; extra == 'test'
 Requires-Dist: docker<8.0.0,>=5.0.2; extra == 'test'
@@ -91,7 +92,7 @@ Requires-Dist: numpy>=1.24.0; extra == 'test'
 Requires-Dist: onnx>=1.15.0; extra == 'test'
 Requires-Dist: pandas<1.5,>=1.3.5; extra == 'test'
 Requires-Dist: pillow<=11,>=10.0.1; extra == 'test'
-Requires-Dist: pyspark==3.3.1; extra == 'test'
+Requires-Dist: pyspark==3.3.2; extra == 'test'
 Requires-Dist: pytest-cov==3.0.0; extra == 'test'
 Requires-Dist: pytest-rerunfailures==10.2; extra == 'test'
 Requires-Dist: pytest-timeout==2.1.0; extra == 'test'

sagemaker-2.232.3/VERSION

@@ -0,0 +1 @@
+2.232.3

{sagemaker-2.232.1 → sagemaker-2.232.3}/pyproject.toml

@@ -48,6 +48,7 @@ dependencies = [
   "PyYAML~=6.0",
   "requests",
   "sagemaker-core>=1.0.0,<2.0.0",
+  "sagemaker-mlflow",
   "schema",
   "smdebug_rulesconfig==1.0.1",
   "tblib>=1.7.0,<4",

{sagemaker-2.232.1 → sagemaker-2.232.3}/requirements/extras/feature-processor_requirements.txt

@@ -1,2 +1,2 @@
-pyspark==3.3.1
+pyspark==3.3.2
 sagemaker-feature-store-pyspark-3.3

{sagemaker-2.232.1 → sagemaker-2.232.3}/requirements/extras/test_requirements.txt

@@ -24,7 +24,7 @@ Jinja2==3.1.4
 pyvis==0.2.1
 pandas>=1.3.5,<1.5
 scikit-learn==1.3.0
-cloudpickle==2.2.1
+cloudpickle>=2.2.1
 PyYAML==6.0
 # TODO find workaround
 xgboost>=1.6.2,<=1.7.6

{sagemaker-2.232.1 → sagemaker-2.232.3}/src/sagemaker/estimator.py

@@ -107,6 +107,8 @@ from sagemaker.workflow.entities import PipelineVariable
 from sagemaker.workflow.parameters import ParameterString
 from sagemaker.workflow.pipeline_context import PipelineSession, runnable_by_pipeline
 
+from sagemaker.mlflow.forward_sagemaker_metrics import log_sagemaker_job_to_mlflow
+
 logger = logging.getLogger(__name__)
 
 
@@ -1366,8 +1368,14 @@ class EstimatorBase(with_metaclass(ABCMeta, object)):  # pylint: disable=too-man
         experiment_config = check_and_get_run_experiment_config(experiment_config)
         self.latest_training_job = _TrainingJob.start_new(self, inputs, experiment_config)
         self.jobs.append(self.latest_training_job)
+        forward_to_mlflow_tracking_server = False
+        if os.environ.get("MLFLOW_TRACKING_URI") and self.enable_network_isolation():
+            wait = True
+            forward_to_mlflow_tracking_server = True
         if wait:
             self.latest_training_job.wait(logs=logs)
+        if forward_to_mlflow_tracking_server:
+            log_sagemaker_job_to_mlflow(self.latest_training_job.name)
 
     def _compilation_job_name(self):
         """Placeholder docstring"""

{sagemaker-2.232.1 → sagemaker-2.232.3}/src/sagemaker/fw_utils.py

@@ -153,6 +153,7 @@ TORCH_DISTRIBUTED_GPU_SUPPORTED_FRAMEWORK_VERSIONS = [
     "2.1.2",
     "2.2.0",
     "2.3.1",
+    "2.4.1",
 ]
 
 TRAINIUM_SUPPORTED_DISTRIBUTION_STRATEGIES = ["torch_distributed"]

{sagemaker-2.232.1 → sagemaker-2.232.3}/src/sagemaker/huggingface/model.py

@@ -334,11 +334,7 @@ class HuggingFaceModel(FrameworkModel):
             container_startup_health_check_timeout=container_startup_health_check_timeout,
             inference_recommendation_id=inference_recommendation_id,
             explainer_config=explainer_config,
-            endpoint_logging=kwargs.get("endpoint_logging", False),
-            endpoint_type=kwargs.get("endpoint_type", None),
-            resources=kwargs.get("resources", None),
-            managed_instance_scaling=kwargs.get("managed_instance_scaling", None),
-            routing_config=kwargs.get("routing_config", None),
+            **kwargs,
         )
 
     def register(

{sagemaker-2.232.1 → sagemaker-2.232.3}/src/sagemaker/image_uri_config/djl-lmi.json

@@ -28,6 +28,7 @@
                 "eu-west-3": "763104351884",
                 "eu-south-1": "692866216735",
                 "me-south-1": "217643126080",
+                "me-central-1": "914824155844",
                 "sa-east-1": "763104351884",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",
@@ -62,6 +63,7 @@
                 "eu-west-3": "763104351884",
                 "eu-south-1": "692866216735",
                 "me-south-1": "217643126080",
+                "me-central-1": "914824155844",
                 "sa-east-1": "763104351884",
                 "us-east-1": "763104351884",
                 "us-east-2": "763104351884",

{sagemaker-2.232.1 → sagemaker-2.232.3}/src/sagemaker/image_uri_config/djl-neuronx.json

@@ -15,6 +15,7 @@
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
+                "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
                 "sa-east-1": "763104351884",
@@ -37,6 +38,7 @@
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
+                "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
                 "sa-east-1": "763104351884",
@@ -59,6 +61,7 @@
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
+                "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
                 "sa-east-1": "763104351884",
@@ -81,6 +84,7 @@
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
+                "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
                 "sa-east-1": "763104351884",
@@ -103,6 +107,7 @@
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
+                "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
                 "sa-east-1": "763104351884",
@@ -125,6 +130,7 @@
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
+                "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
                 "sa-east-1": "763104351884",
@@ -147,6 +153,7 @@
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
+                "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
                 "sa-east-1": "763104351884",
@@ -169,6 +176,7 @@
                 "cn-north-1": "727897471807",
                 "cn-northwest-1": "727897471807",
                 "eu-central-1": "763104351884",
+                "eu-central-2": "380420809688",
                 "eu-west-1": "763104351884",
                 "eu-west-3": "763104351884",
                 "sa-east-1": "763104351884",

{sagemaker-2.232.1 → sagemaker-2.232.3}/src/sagemaker/image_uri_config/huggingface-llm-neuronx.json

@@ -4,7 +4,7 @@
             "inf2"
         ],
         "version_aliases": {
-            "0.0": "0.0.24"
+            "0.0": "0.0.25"
         },
         "versions": {
             "0.0.16": {
@@ -301,6 +301,39 @@
                 "container_version": {
                     "inf2": "ubuntu22.04"
                 }
+            },
+            "0.0.25": {
+                "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",
+                    "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.25",
+                "repository": "huggingface-pytorch-tgi-inference",
+                "container_version": {
+                    "inf2": "ubuntu22.04"
+                }
             }
         }
     }

{sagemaker-2.232.1 → sagemaker-2.232.3}/src/sagemaker/image_uri_config/huggingface-llm.json

@@ -12,7 +12,7 @@
             "1.2": "1.2.0",
             "1.3": "1.3.3",
             "1.4": "1.4.5",
-            "2.0": "2.2.0"
+            "2.0": "2.3.1"
         },
         "versions": {
             "0.6.0": {
@@ -719,6 +719,53 @@
                 "container_version": {
                     "gpu": "cu121-ubuntu22.04-v2.0"
                 }
+            },
+            "2.3.1": {
+                "py_versions": [
+                    "py311"
+                ],
+                "registries": {
+                    "af-south-1": "626614931356",
+                    "il-central-1": "780543022126",
+                    "ap-east-1": "871362719292",
+                    "ap-northeast-1": "763104351884",
+                    "ap-northeast-2": "763104351884",
+                    "ap-northeast-3": "364406365360",
+                    "ap-south-1": "763104351884",
+                    "ap-south-2": "772153158452",
+                    "ap-southeast-1": "763104351884",
+                    "ap-southeast-2": "763104351884",
+                    "ap-southeast-3": "907027046896",
+                    "ap-southeast-4": "457447274322",
+                    "ca-central-1": "763104351884",
+                    "cn-north-1": "727897471807",
+                    "cn-northwest-1": "727897471807",
+                    "eu-central-1": "763104351884",
+                    "eu-central-2": "380420809688",
+                    "eu-north-1": "763104351884",
+                    "eu-west-1": "763104351884",
+                    "eu-west-2": "763104351884",
+                    "eu-west-3": "763104351884",      
+                    "eu-south-1": "692866216735",
+                    "eu-south-2": "503227376785",
+                    "me-south-1": "217643126080",
+                    "me-central-1": "914824155844",
+                    "sa-east-1": "763104351884",
+                    "us-east-1": "763104351884",
+                    "us-east-2": "763104351884",
+                    "us-gov-east-1": "446045086412",
+                    "us-gov-west-1": "442386744353",
+                    "us-iso-east-1": "886529160074",
+                    "us-isob-east-1": "094389454867",
+                    "us-west-1": "763104351884",
+                    "us-west-2": "763104351884",
+                    "ca-west-1": "204538143572"
+                },
+                "tag_prefix": "2.4.0-tgi2.3.1",
+                "repository": "huggingface-pytorch-tgi-inference",
+                "container_version": {
+                    "gpu": "cu124-ubuntu22.04"
+                }
             }
         }
     }

{sagemaker-2.232.1 → sagemaker-2.232.3}/src/sagemaker/image_uri_config/image-classification-neo.json

@@ -1,5 +1,7 @@
 {
-    "scope": ["inference"],
+    "scope": [
+        "inference"
+    ],
     "versions": {
         "latest": {
             "registries": {
@@ -15,11 +17,13 @@
                 "cn-north-1": "472730292857",
                 "cn-northwest-1": "474822919863",
                 "eu-central-1": "746233611703",
+                "eu-central-2": "010526262399",
                 "eu-north-1": "601324751636",
                 "eu-south-1": "966458181534",
                 "eu-west-1": "802834080501",
                 "eu-west-2": "205493899709",
                 "eu-west-3": "254080097072",
+                "il-central-1": "275950707576",
                 "me-south-1": "836785723513",
                 "sa-east-1": "756306329178",
                 "us-east-1": "785573368785",
@@ -27,11 +31,12 @@
                 "us-gov-west-1": "263933020539",
                 "us-iso-east-1": "167761179201",
                 "us-isob-east-1": "406031935815",
+                "us-isof-east-1": "751086301963",
+                "us-isof-south-1": "935523707064",
                 "us-west-1": "710691900526",
-                "us-west-2": "301217895009",
-                "il-central-1": "275950707576"
+                "us-west-2": "301217895009"
             },
             "repository": "image-classification-neo"
         }
     }
-}
+}

{sagemaker-2.232.1 → sagemaker-2.232.3}/src/sagemaker/image_uri_config/inferentia-mxnet.json

@@ -1,9 +1,15 @@
 {
-    "processors": ["inf"],
-    "scope": ["inference"],
+    "processors": [
+        "inf"
+    ],
+    "scope": [
+        "inference"
+    ],
     "versions": {
         "1.5.1": {
-            "py_versions": ["py3"],
+            "py_versions": [
+                "py3"
+            ],
             "registries": {
                 "af-south-1": "774647643957",
                 "ap-east-1": "110948597952",
@@ -17,11 +23,13 @@
                 "cn-north-1": "472730292857",
                 "cn-northwest-1": "474822919863",
                 "eu-central-1": "746233611703",
+                "eu-central-2": "010526262399",
                 "eu-north-1": "601324751636",
                 "eu-south-1": "966458181534",
                 "eu-west-1": "802834080501",
                 "eu-west-2": "205493899709",
                 "eu-west-3": "254080097072",
+                "il-central-1": "275950707576",
                 "me-south-1": "836785723513",
                 "sa-east-1": "756306329178",
                 "us-east-1": "785573368785",
@@ -29,14 +37,17 @@
                 "us-gov-west-1": "263933020539",
                 "us-iso-east-1": "167761179201",
                 "us-isob-east-1": "406031935815",
+                "us-isof-east-1": "751086301963",
+                "us-isof-south-1": "935523707064",
                 "us-west-1": "710691900526",
-                "us-west-2": "301217895009",
-                "il-central-1": "275950707576"
+                "us-west-2": "301217895009"
             },
             "repository": "sagemaker-neo-mxnet"
         },
         "1.8": {
-            "py_versions": ["py3"],
+            "py_versions": [
+                "py3"
+            ],
             "registries": {
                 "af-south-1": "774647643957",
                 "ap-east-1": "110948597952",
@@ -50,11 +61,13 @@
                 "cn-north-1": "472730292857",
                 "cn-northwest-1": "474822919863",
                 "eu-central-1": "746233611703",
+                "eu-central-2": "010526262399",
                 "eu-north-1": "601324751636",
                 "eu-south-1": "966458181534",
                 "eu-west-1": "802834080501",
                 "eu-west-2": "205493899709",
                 "eu-west-3": "254080097072",
+                "il-central-1": "275950707576",
                 "me-south-1": "836785723513",
                 "sa-east-1": "756306329178",
                 "us-east-1": "785573368785",
@@ -62,11 +75,12 @@
                 "us-gov-west-1": "263933020539",
                 "us-iso-east-1": "167761179201",
                 "us-isob-east-1": "406031935815",
+                "us-isof-east-1": "751086301963",
+                "us-isof-south-1": "935523707064",
                 "us-west-1": "710691900526",
-                "us-west-2": "301217895009",
-                "il-central-1": "275950707576"
+                "us-west-2": "301217895009"
             },
             "repository": "sagemaker-neo-mxnet"
         }
     }
-}
+}

{sagemaker-2.232.1 → sagemaker-2.232.3}/src/sagemaker/image_uri_config/inferentia-pytorch.json

@@ -1,9 +1,15 @@
 {
-    "processors": ["inf"],
-    "scope": ["inference"],
+    "processors": [
+        "inf"
+    ],
+    "scope": [
+        "inference"
+    ],
     "versions": {
         "1.7": {
-            "py_versions": ["py3"],
+            "py_versions": [
+                "py3"
+            ],
             "registries": {
                 "af-south-1": "774647643957",
                 "ap-east-1": "110948597952",
@@ -17,11 +23,13 @@
                 "cn-north-1": "472730292857",
                 "cn-northwest-1": "474822919863",
                 "eu-central-1": "746233611703",
+                "eu-central-2": "010526262399",
                 "eu-north-1": "601324751636",
                 "eu-south-1": "966458181534",
                 "eu-west-1": "802834080501",
                 "eu-west-2": "205493899709",
                 "eu-west-3": "254080097072",
+                "il-central-1": "275950707576",
                 "me-south-1": "836785723513",
                 "sa-east-1": "756306329178",
                 "us-east-1": "785573368785",
@@ -29,14 +37,17 @@
                 "us-gov-west-1": "263933020539",
                 "us-iso-east-1": "167761179201",
                 "us-isob-east-1": "406031935815",
+                "us-isof-east-1": "751086301963",
+                "us-isof-south-1": "935523707064",
                 "us-west-1": "710691900526",
-                "us-west-2": "301217895009",
-                "il-central-1": "275950707576"
+                "us-west-2": "301217895009"
             },
             "repository": "sagemaker-neo-pytorch"
         },
         "1.8": {
-            "py_versions": ["py3"],
+            "py_versions": [
+                "py3"
+            ],
             "registries": {
                 "af-south-1": "774647643957",
                 "ap-east-1": "110948597952",
@@ -50,11 +61,13 @@
                 "cn-north-1": "472730292857",
                 "cn-northwest-1": "474822919863",
                 "eu-central-1": "746233611703",
+                "eu-central-2": "010526262399",
                 "eu-north-1": "601324751636",
                 "eu-south-1": "966458181534",
                 "eu-west-1": "802834080501",
                 "eu-west-2": "205493899709",
                 "eu-west-3": "254080097072",
+                "il-central-1": "275950707576",
                 "me-south-1": "836785723513",
                 "sa-east-1": "756306329178",
                 "us-east-1": "785573368785",
@@ -62,14 +75,17 @@
                 "us-gov-west-1": "263933020539",
                 "us-iso-east-1": "167761179201",
                 "us-isob-east-1": "406031935815",
+                "us-isof-east-1": "751086301963",
+                "us-isof-south-1": "935523707064",
                 "us-west-1": "710691900526",
-                "us-west-2": "301217895009",
-                "il-central-1": "275950707576"
+                "us-west-2": "301217895009"
             },
             "repository": "sagemaker-neo-pytorch"
         },
         "1.9": {
-            "py_versions": ["py3"],
+            "py_versions": [
+                "py3"
+            ],
             "registries": {
                 "af-south-1": "774647643957",
                 "ap-east-1": "110948597952",
@@ -83,11 +99,13 @@
                 "cn-north-1": "472730292857",
                 "cn-northwest-1": "474822919863",
                 "eu-central-1": "746233611703",
+                "eu-central-2": "010526262399",
                 "eu-north-1": "601324751636",
                 "eu-south-1": "966458181534",
                 "eu-west-1": "802834080501",
                 "eu-west-2": "205493899709",
                 "eu-west-3": "254080097072",
+                "il-central-1": "275950707576",
                 "me-south-1": "836785723513",
                 "sa-east-1": "756306329178",
                 "us-east-1": "785573368785",
@@ -95,11 +113,12 @@
                 "us-gov-west-1": "263933020539",
                 "us-iso-east-1": "167761179201",
                 "us-isob-east-1": "406031935815",
+                "us-isof-east-1": "751086301963",
+                "us-isof-south-1": "935523707064",
                 "us-west-1": "710691900526",
-                "us-west-2": "301217895009",
-                "il-central-1": "275950707576"
+                "us-west-2": "301217895009"
             },
             "repository": "sagemaker-neo-pytorch"
         }
     }
-}
+}

{sagemaker-2.232.1 → sagemaker-2.232.3}/src/sagemaker/image_uri_config/inferentia-tensorflow.json

@@ -1,9 +1,15 @@
 {
-    "processors": ["inf"],
-    "scope": ["inference"],
+    "processors": [
+        "inf"
+    ],
+    "scope": [
+        "inference"
+    ],
     "versions": {
         "1.15.0": {
-            "py_versions": ["py3"],
+            "py_versions": [
+                "py3"
+            ],
             "registries": {
                 "af-south-1": "774647643957",
                 "ap-east-1": "110948597952",
@@ -17,11 +23,13 @@
                 "cn-north-1": "472730292857",
                 "cn-northwest-1": "474822919863",
                 "eu-central-1": "746233611703",
+                "eu-central-2": "010526262399",
                 "eu-north-1": "601324751636",
                 "eu-south-1": "966458181534",
                 "eu-west-1": "802834080501",
                 "eu-west-2": "205493899709",
                 "eu-west-3": "254080097072",
+                "il-central-1": "275950707576",
                 "me-south-1": "836785723513",
                 "sa-east-1": "756306329178",
                 "us-east-1": "785573368785",
@@ -29,14 +37,17 @@
                 "us-gov-west-1": "263933020539",
                 "us-iso-east-1": "167761179201",
                 "us-isob-east-1": "406031935815",
+                "us-isof-east-1": "751086301963",
+                "us-isof-south-1": "935523707064",
                 "us-west-1": "710691900526",
-                "us-west-2": "301217895009",
-                "il-central-1": "275950707576"
+                "us-west-2": "301217895009"
             },
             "repository": "sagemaker-neo-tensorflow"
         },
         "2.5.2": {
-            "py_versions": ["py3"],
+            "py_versions": [
+                "py3"
+            ],
             "registries": {
                 "af-south-1": "774647643957",
                 "ap-east-1": "110948597952",
@@ -50,11 +61,13 @@
                 "cn-north-1": "472730292857",
                 "cn-northwest-1": "474822919863",
                 "eu-central-1": "746233611703",
+                "eu-central-2": "010526262399",
                 "eu-north-1": "601324751636",
                 "eu-south-1": "966458181534",
                 "eu-west-1": "802834080501",
                 "eu-west-2": "205493899709",
                 "eu-west-3": "254080097072",
+                "il-central-1": "275950707576",
                 "me-south-1": "836785723513",
                 "sa-east-1": "756306329178",
                 "us-east-1": "785573368785",
@@ -62,11 +75,12 @@
                 "us-gov-west-1": "263933020539",
                 "us-iso-east-1": "167761179201",
                 "us-isob-east-1": "406031935815",
+                "us-isof-east-1": "751086301963",
+                "us-isof-south-1": "935523707064",
                 "us-west-1": "710691900526",
-                "us-west-2": "301217895009",
-                "il-central-1": "275950707576"
+                "us-west-2": "301217895009"
             },
             "repository": "sagemaker-neo-tensorflow"
         }
     }
-}
+}