From fd79ef0644f8045780924fe4e2d49bb6b99af190 Mon Sep 17 00:00:00 2001 From: Tony Hu Date: Wed, 22 Mar 2023 16:34:15 -0700 Subject: [PATCH 1/2] feature: AutoGluon 0.7.0 image_uris update --- src/sagemaker/image_uri_config/autogluon.json | 73 ++++++++++++++++++- .../sagemaker/image_uris/test_autogluon.py | 2 + 2 files changed, 74 insertions(+), 1 deletion(-) diff --git a/src/sagemaker/image_uri_config/autogluon.json b/src/sagemaker/image_uri_config/autogluon.json index 230e2a6951..ef64f06d50 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"] } } }, @@ -623,6 +658,42 @@ "repository": "autogluon-inference", "processors": ["cpu", "gpu"], "py_versions": ["py38"] + }, + "0.6.2": { + "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..e62be32a3b 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"] From 453807dbd06529972e35aa641761d82c8ac63f63 Mon Sep 17 00:00:00 2001 From: Tony Hu Date: Wed, 22 Mar 2023 21:36:40 -0700 Subject: [PATCH 2/2] fixed tests and added 0.7 to inference --- src/sagemaker/image_uri_config/autogluon.json | 5 +++-- tests/unit/sagemaker/image_uris/test_autogluon.py | 7 ++++++- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/src/sagemaker/image_uri_config/autogluon.json b/src/sagemaker/image_uri_config/autogluon.json index ef64f06d50..4ce304d59d 100644 --- a/src/sagemaker/image_uri_config/autogluon.json +++ b/src/sagemaker/image_uri_config/autogluon.json @@ -344,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": { @@ -659,7 +660,7 @@ "processors": ["cpu", "gpu"], "py_versions": ["py38"] }, - "0.6.2": { + "0.7.0": { "registries": { "af-south-1": "626614931356", "ap-east-1": "871362719292", diff --git a/tests/unit/sagemaker/image_uris/test_autogluon.py b/tests/unit/sagemaker/image_uris/test_autogluon.py index e62be32a3b..f668eaf5a8 100644 --- a/tests/unit/sagemaker/image_uris/test_autogluon.py +++ b/tests/unit/sagemaker/image_uris/test_autogluon.py @@ -70,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"