diff --git a/src/sagemaker/image_uri_config/autogluon.json b/src/sagemaker/image_uri_config/autogluon.json index 230e2a6951..4ce304d59d 100644 --- a/src/sagemaker/image_uri_config/autogluon.json +++ b/src/sagemaker/image_uri_config/autogluon.json @@ -5,7 +5,8 @@ "0.3": "0.3.2", "0.4": "0.4.3", "0.5": "0.5.2", - "0.6": "0.6.2" + "0.6": "0.6.2", + "0.7": "0.7.0" }, "versions": { "0.3.1": { @@ -301,6 +302,40 @@ "repository": "autogluon-training", "processors": ["cpu", "gpu"], "py_versions": ["py38"] + }, + "0.7.0": { + "registries": { + "af-south-1": "626614931356", + "ap-east-1": "871362719292", + "ap-northeast-1": "763104351884", + "ap-northeast-2": "763104351884", + "ap-northeast-3": "364406365360", + "ap-south-1": "763104351884", + "ap-southeast-1": "763104351884", + "ap-southeast-2": "763104351884", + "ap-southeast-3": "907027046896", + "ap-southeast-4": "457447274322", + "ca-central-1": "763104351884", + "eu-central-1": "763104351884", + "eu-north-1": "763104351884", + "eu-west-1": "763104351884", + "eu-west-2": "763104351884", + "eu-west-3": "763104351884", + "eu-south-1": "692866216735", + "me-south-1": "217643126080", + "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" + }, + "repository": "autogluon-training", + "processors": ["cpu", "gpu"], + "py_versions": ["py39"] } } }, @@ -309,7 +344,8 @@ "0.3": "0.3.2", "0.4": "0.4.3", "0.5": "0.5.2", - "0.6": "0.6.2" + "0.6": "0.6.2", + "0.7": "0.7.0" }, "versions": { "0.3.1": { @@ -623,6 +659,42 @@ "repository": "autogluon-inference", "processors": ["cpu", "gpu"], "py_versions": ["py38"] + }, + "0.7.0": { + "registries": { + "af-south-1": "626614931356", + "ap-east-1": "871362719292", + "ap-northeast-1": "763104351884", + "ap-northeast-2": "763104351884", + "ap-northeast-3": "364406365360", + "ap-south-1": "763104351884", + "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-north-1": "763104351884", + "eu-west-1": "763104351884", + "eu-west-2": "763104351884", + "eu-west-3": "763104351884", + "eu-south-1": "692866216735", + "me-south-1": "217643126080", + "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" + }, + "repository": "autogluon-inference", + "processors": ["cpu", "gpu"], + "py_versions": ["py39"] } } } diff --git a/tests/unit/sagemaker/image_uris/test_autogluon.py b/tests/unit/sagemaker/image_uris/test_autogluon.py index a51f36b7b0..f668eaf5a8 100644 --- a/tests/unit/sagemaker/image_uris/test_autogluon.py +++ b/tests/unit/sagemaker/image_uris/test_autogluon.py @@ -57,6 +57,8 @@ "0.6.1", "0.6", "0.6.2", + "0.7.0", + "0.7", ] SCOPES = ["training", "inference"] @@ -68,7 +70,12 @@ @pytest.mark.parametrize("processor", PROCESSORS) def test_valid_uris_training(version, scope, processor): instance_type = "ml.c4.xlarge" if processor == "cpu" else "ml.p2.xlarge" - py_version = "py37" if version == "0.3.1" else "py38" + if version == "0.3.1": + py_version = "py37" + elif version < "0.7": + py_version = "py38" + else: + py_version = "py39" if ( scope == "inference" and processor == "gpu"