fixed tests and added 0.7 to inference

Author: tonyhoo

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",

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"