Skip to content

Commit 20ae406

Browse files
gradientskyJoseJuan98
gradientsky
authored and
JoseJuan98
committed
feature: AutoGluon 0.4.3 and 0.5.2 image_uris (aws#3274)
1 parent ead5c4e commit 20ae406

File tree

2 files changed

+131
-3
lines changed

2 files changed

+131
-3
lines changed

‎src/sagemaker/image_uri_config/autogluon.json

Lines changed: 130 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33
"processors": ["cpu", "gpu"],
44
"version_aliases": {
55
"0.3": "0.3.2",
6-
"0.4": "0.4.2"
6+
"0.4": "0.4.3",
7+
"0.5": "0.5.2"
78
},
89
"versions": {
910
"0.3.1": {
@@ -125,13 +126,74 @@
125126
},
126127
"repository": "autogluon-training",
127128
"py_versions": ["py38"]
129+
},
130+
"0.4.3": {
131+
"registries": {
132+
"af-south-1": "626614931356",
133+
"ap-east-1": "871362719292",
134+
"ap-northeast-1": "763104351884",
135+
"ap-northeast-2": "763104351884",
136+
"ap-northeast-3": "364406365360",
137+
"ap-south-1": "763104351884",
138+
"ap-southeast-1": "763104351884",
139+
"ap-southeast-2": "763104351884",
140+
"ap-southeast-3": "907027046896",
141+
"ca-central-1": "763104351884",
142+
"eu-central-1": "763104351884",
143+
"eu-north-1": "763104351884",
144+
"eu-west-1": "763104351884",
145+
"eu-west-2": "763104351884",
146+
"eu-west-3": "763104351884",
147+
"eu-south-1": "692866216735",
148+
"me-south-1": "217643126080",
149+
"sa-east-1": "763104351884",
150+
"us-east-1": "763104351884",
151+
"us-east-2": "763104351884",
152+
"us-gov-west-1": "442386744353",
153+
"us-iso-east-1": "886529160074",
154+
"us-west-1": "763104351884",
155+
"us-west-2": "763104351884"
156+
},
157+
"repository": "autogluon-training",
158+
"py_versions": ["py38"]
159+
},
160+
"0.5.2": {
161+
"registries": {
162+
"af-south-1": "626614931356",
163+
"ap-east-1": "871362719292",
164+
"ap-northeast-1": "763104351884",
165+
"ap-northeast-2": "763104351884",
166+
"ap-northeast-3": "364406365360",
167+
"ap-south-1": "763104351884",
168+
"ap-southeast-1": "763104351884",
169+
"ap-southeast-2": "763104351884",
170+
"ap-southeast-3": "907027046896",
171+
"ca-central-1": "763104351884",
172+
"eu-central-1": "763104351884",
173+
"eu-north-1": "763104351884",
174+
"eu-west-1": "763104351884",
175+
"eu-west-2": "763104351884",
176+
"eu-west-3": "763104351884",
177+
"eu-south-1": "692866216735",
178+
"me-south-1": "217643126080",
179+
"sa-east-1": "763104351884",
180+
"us-east-1": "763104351884",
181+
"us-east-2": "763104351884",
182+
"us-gov-west-1": "442386744353",
183+
"us-iso-east-1": "886529160074",
184+
"us-west-1": "763104351884",
185+
"us-west-2": "763104351884"
186+
},
187+
"repository": "autogluon-training",
188+
"py_versions": ["py38"]
128189
}
129190
}
130191
},
131192
"inference": {
132193
"version_aliases": {
133194
"0.3": "0.3.2",
134-
"0.4": "0.4.2"
195+
"0.4": "0.4.3",
196+
"0.5": "0.5.2"
135197
},
136198
"versions": {
137199
"0.3.1": {
@@ -265,6 +327,72 @@
265327
"repository": "autogluon-inference",
266328
"processors": ["cpu", "gpu"],
267329
"py_versions": ["py38"]
330+
},
331+
"0.4.3": {
332+
"registries": {
333+
"af-south-1": "626614931356",
334+
"ap-east-1": "871362719292",
335+
"ap-northeast-1": "763104351884",
336+
"ap-northeast-2": "763104351884",
337+
"ap-northeast-3": "364406365360",
338+
"ap-south-1": "763104351884",
339+
"ap-southeast-1": "763104351884",
340+
"ap-southeast-2": "763104351884",
341+
"ap-southeast-3": "907027046896",
342+
"ca-central-1": "763104351884",
343+
"cn-north-1": "727897471807",
344+
"cn-northwest-1": "727897471807",
345+
"eu-central-1": "763104351884",
346+
"eu-north-1": "763104351884",
347+
"eu-west-1": "763104351884",
348+
"eu-west-2": "763104351884",
349+
"eu-west-3": "763104351884",
350+
"eu-south-1": "692866216735",
351+
"me-south-1": "217643126080",
352+
"sa-east-1": "763104351884",
353+
"us-east-1": "763104351884",
354+
"us-east-2": "763104351884",
355+
"us-gov-west-1": "442386744353",
356+
"us-iso-east-1": "886529160074",
357+
"us-west-1": "763104351884",
358+
"us-west-2": "763104351884"
359+
},
360+
"repository": "autogluon-inference",
361+
"processors": ["cpu", "gpu"],
362+
"py_versions": ["py38"]
363+
},
364+
"0.5.2": {
365+
"registries": {
366+
"af-south-1": "626614931356",
367+
"ap-east-1": "871362719292",
368+
"ap-northeast-1": "763104351884",
369+
"ap-northeast-2": "763104351884",
370+
"ap-northeast-3": "364406365360",
371+
"ap-south-1": "763104351884",
372+
"ap-southeast-1": "763104351884",
373+
"ap-southeast-2": "763104351884",
374+
"ap-southeast-3": "907027046896",
375+
"ca-central-1": "763104351884",
376+
"cn-north-1": "727897471807",
377+
"cn-northwest-1": "727897471807",
378+
"eu-central-1": "763104351884",
379+
"eu-north-1": "763104351884",
380+
"eu-west-1": "763104351884",
381+
"eu-west-2": "763104351884",
382+
"eu-west-3": "763104351884",
383+
"eu-south-1": "692866216735",
384+
"me-south-1": "217643126080",
385+
"sa-east-1": "763104351884",
386+
"us-east-1": "763104351884",
387+
"us-east-2": "763104351884",
388+
"us-gov-west-1": "442386744353",
389+
"us-iso-east-1": "886529160074",
390+
"us-west-1": "763104351884",
391+
"us-west-2": "763104351884"
392+
},
393+
"repository": "autogluon-inference",
394+
"processors": ["cpu", "gpu"],
395+
"py_versions": ["py38"]
268396
}
269397
}
270398
}

‎tests/unit/sagemaker/image_uris/test_autogluon.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@
4242
"us-west-1": "763104351884",
4343
"us-west-2": "763104351884",
4444
}
45-
VERSIONS = ["0.3.1", "0.3.2", "0.4.0", "0.4.2", "0.3", "0.4"]
45+
VERSIONS = ["0.3.1", "0.3.2", "0.4.0", "0.4.2", "0.4.3", "0.3", "0.4", "0.5.2", "0.5"]
4646
SCOPES = ["training", "inference"]
4747
PROCESSORS = ["cpu", "gpu"]
4848

0 commit comments

Comments
 (0)