feature: AutoGluon 0.3.2 and 0.4.0 image_uris (aws#2991)

Author: gradientsky

src/sagemaker/image_uri_config/autogluon.json

@@ -1,6 +1,10 @@
 {
     "training": {
         "processors": ["cpu", "gpu"],
+        "version_aliases": {
+            "0.3": "0.3.2",
+            "0.4": "0.4.0"
+        },
         "versions": {
             "0.3.1": {
                 "registries": {
@@ -30,11 +34,73 @@
                 },
                 "repository": "autogluon-training",
                 "py_versions": ["py37"]
+            },
+            "0.3.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",
+                    "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-west-1": "442386744353",
+                    "us-iso-east-1": "886529160074",
+                    "us-west-1": "763104351884",
+                    "us-west-2": "763104351884"
+                },
+                "repository": "autogluon-training",
+                "py_versions": ["py38"]
+            },
+            "0.4.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",
+                    "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-west-1": "442386744353",
+                    "us-iso-east-1": "886529160074",
+                    "us-west-1": "763104351884",
+                    "us-west-2": "763104351884"
+                },
+                "repository": "autogluon-training",
+                "py_versions": ["py38"]
             }
         }
     },
     "inference": {
         "processors": ["cpu"],
+        "version_aliases": {
+            "0.3": "0.3.2",
+            "0.4": "0.4.0"
+        },
         "versions": {
             "0.3.1": {
                 "registries": {
@@ -66,6 +132,68 @@
                 },
                 "repository": "autogluon-inference",
                 "py_versions": ["py37"]
+            },
+            "0.3.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",
+                    "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-west-1": "442386744353",
+                    "us-iso-east-1": "886529160074",
+                    "us-west-1": "763104351884",
+                    "us-west-2": "763104351884"
+                },
+                "repository": "autogluon-inference",
+                "py_versions": ["py38"]
+            },
+            "0.4.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",
+                    "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-west-1": "442386744353",
+                    "us-iso-east-1": "886529160074",
+                    "us-west-1": "763104351884",
+                    "us-west-2": "763104351884"
+                },
+                "repository": "autogluon-inference",
+                "py_versions": ["py38"]
             }
         }
     }

tests/unit/sagemaker/image_uris/test_autogluon.py

@@ -42,17 +42,18 @@
     "us-west-1": "763104351884",
     "us-west-2": "763104351884",
 }
-VERSIONS = ["0.3.1"]
+VERSIONS = ["0.3.1", "0.3.2", "0.4.0", "0.3", "0.4"]
 
 
 @pytest.mark.parametrize("version", VERSIONS)
 def test_valid_uris(version):
+    py_version = "py37" if version == "0.3.1" else "py38"
     for region in ACCOUNTS.keys():
         uri = image_uris.retrieve(
             "autogluon",
             region=region,
             version=version,
-            py_version="py37",
+            py_version=py_version,
             image_scope="training",
             instance_type="ml.c4.xlarge",
         )
@@ -61,7 +62,7 @@ def test_valid_uris(version):
             "autogluon-training",
             version,
             ACCOUNTS[region],
-            py_version="py37",
+            py_version=py_version,
             region=region,
         )
         assert uri == expected