Skip to content

Commit 9946d67

Browse files
authored
feature: AutoGluon 0.7.0 image_uris update (#3746)
1 parent 6e5cd23 commit 9946d67

File tree

2 files changed

+82
-3
lines changed

2 files changed

+82
-3
lines changed

src/sagemaker/image_uri_config/autogluon.json

+74-2
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@
55
"0.3": "0.3.2",
66
"0.4": "0.4.3",
77
"0.5": "0.5.2",
8-
"0.6": "0.6.2"
8+
"0.6": "0.6.2",
9+
"0.7": "0.7.0"
910
},
1011
"versions": {
1112
"0.3.1": {
@@ -301,6 +302,40 @@
301302
"repository": "autogluon-training",
302303
"processors": ["cpu", "gpu"],
303304
"py_versions": ["py38"]
305+
},
306+
"0.7.0": {
307+
"registries": {
308+
"af-south-1": "626614931356",
309+
"ap-east-1": "871362719292",
310+
"ap-northeast-1": "763104351884",
311+
"ap-northeast-2": "763104351884",
312+
"ap-northeast-3": "364406365360",
313+
"ap-south-1": "763104351884",
314+
"ap-southeast-1": "763104351884",
315+
"ap-southeast-2": "763104351884",
316+
"ap-southeast-3": "907027046896",
317+
"ap-southeast-4": "457447274322",
318+
"ca-central-1": "763104351884",
319+
"eu-central-1": "763104351884",
320+
"eu-north-1": "763104351884",
321+
"eu-west-1": "763104351884",
322+
"eu-west-2": "763104351884",
323+
"eu-west-3": "763104351884",
324+
"eu-south-1": "692866216735",
325+
"me-south-1": "217643126080",
326+
"sa-east-1": "763104351884",
327+
"us-east-1": "763104351884",
328+
"us-east-2": "763104351884",
329+
"us-gov-east-1": "446045086412",
330+
"us-gov-west-1": "442386744353",
331+
"us-iso-east-1": "886529160074",
332+
"us-isob-east-1": "094389454867",
333+
"us-west-1": "763104351884",
334+
"us-west-2": "763104351884"
335+
},
336+
"repository": "autogluon-training",
337+
"processors": ["cpu", "gpu"],
338+
"py_versions": ["py39"]
304339
}
305340
}
306341
},
@@ -309,7 +344,8 @@
309344
"0.3": "0.3.2",
310345
"0.4": "0.4.3",
311346
"0.5": "0.5.2",
312-
"0.6": "0.6.2"
347+
"0.6": "0.6.2",
348+
"0.7": "0.7.0"
313349
},
314350
"versions": {
315351
"0.3.1": {
@@ -623,6 +659,42 @@
623659
"repository": "autogluon-inference",
624660
"processors": ["cpu", "gpu"],
625661
"py_versions": ["py38"]
662+
},
663+
"0.7.0": {
664+
"registries": {
665+
"af-south-1": "626614931356",
666+
"ap-east-1": "871362719292",
667+
"ap-northeast-1": "763104351884",
668+
"ap-northeast-2": "763104351884",
669+
"ap-northeast-3": "364406365360",
670+
"ap-south-1": "763104351884",
671+
"ap-southeast-1": "763104351884",
672+
"ap-southeast-2": "763104351884",
673+
"ap-southeast-3": "907027046896",
674+
"ap-southeast-4": "457447274322",
675+
"ca-central-1": "763104351884",
676+
"cn-north-1": "727897471807",
677+
"cn-northwest-1": "727897471807",
678+
"eu-central-1": "763104351884",
679+
"eu-north-1": "763104351884",
680+
"eu-west-1": "763104351884",
681+
"eu-west-2": "763104351884",
682+
"eu-west-3": "763104351884",
683+
"eu-south-1": "692866216735",
684+
"me-south-1": "217643126080",
685+
"sa-east-1": "763104351884",
686+
"us-east-1": "763104351884",
687+
"us-east-2": "763104351884",
688+
"us-gov-east-1": "446045086412",
689+
"us-gov-west-1": "442386744353",
690+
"us-iso-east-1": "886529160074",
691+
"us-isob-east-1": "094389454867",
692+
"us-west-1": "763104351884",
693+
"us-west-2": "763104351884"
694+
},
695+
"repository": "autogluon-inference",
696+
"processors": ["cpu", "gpu"],
697+
"py_versions": ["py39"]
626698
}
627699
}
628700
}

tests/unit/sagemaker/image_uris/test_autogluon.py

+8-1
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,8 @@
5757
"0.6.1",
5858
"0.6",
5959
"0.6.2",
60+
"0.7.0",
61+
"0.7",
6062
]
6163

6264
SCOPES = ["training", "inference"]
@@ -68,7 +70,12 @@
6870
@pytest.mark.parametrize("processor", PROCESSORS)
6971
def test_valid_uris_training(version, scope, processor):
7072
instance_type = "ml.c4.xlarge" if processor == "cpu" else "ml.p2.xlarge"
71-
py_version = "py37" if version == "0.3.1" else "py38"
73+
if version == "0.3.1":
74+
py_version = "py37"
75+
elif version < "0.7":
76+
py_version = "py38"
77+
else:
78+
py_version = "py39"
7279
if (
7380
scope == "inference"
7481
and processor == "gpu"

0 commit comments

Comments
 (0)