diff --git a/src/sagemaker/image_uri_config/sagemaker-distribution.json b/src/sagemaker/image_uri_config/sagemaker-distribution.json index d9ffca5d7b..9853eb01ae 100644 --- a/src/sagemaker/image_uri_config/sagemaker-distribution.json +++ b/src/sagemaker/image_uri_config/sagemaker-distribution.json @@ -2,10 +2,10 @@ "processors": ["cpu", "gpu"], "scope": ["inference"], "version_aliases": { - "3.0": "3.0.0" + "3.2": "3.2.0" }, "versions": { - "3.0.0": { + "3.2.0": { "registries": { "us-east-1": "885854791233", "us-east-2": "137914896644", diff --git a/tests/unit/sagemaker/image_uris/test_sagemaker_distribution.py b/tests/unit/sagemaker/image_uris/test_sagemaker_distribution.py index d339a50b2e..adc51064f1 100644 --- a/tests/unit/sagemaker/image_uris/test_sagemaker_distribution.py +++ b/tests/unit/sagemaker/image_uris/test_sagemaker_distribution.py @@ -41,7 +41,7 @@ def test_sagemaker_distribution_ecr_uri(load_config): account=SAGEMAKER_DISTRIBUTION_ACCOUNTS[region], region=region, version=version, - tag="3.0.0", + tag="3.2.0", instance_type=INSTANCE_TYPES[processor], processor=processor, )