Skip to content

Commit 48c13e9

Browse files
gradientskyNivekNeyandre-marcos-perezmufaddal-rohawala
authored
feature: AutoGluon 0.6.1 image_uris (#3544)
* fix: type hint of PySparkProcessor __init__ (#3297) * fix: fix PySparkProcessor __init__ params type (#3354) * fix: Revert "fix: type hint of PySparkProcessor __init__" (#3524) * fix: type hint of PySparkProcessor __init__ (#3297) * fix: fix PySparkProcessor __init__ params type (#3354) * fix: Revert "fix: type hint of PySparkProcessor __init__" (#3524) * feature: AutoGluon 0.6.1 image_uris * feature: AutoGluon 0.6.1 image_uris * Added processors for training images for consistency * Formatting * Formatting Co-authored-by: Kevin <[email protected]> Co-authored-by: André Perez <[email protected]> Co-authored-by: Mufaddal Rohawala <[email protected]>
1 parent 047d062 commit 48c13e9

File tree

2 files changed

+88
-3
lines changed

2 files changed

+88
-3
lines changed

‎src/sagemaker/image_uri_config/autogluon.json

+74-2
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,8 @@
44
"version_aliases": {
55
"0.3": "0.3.2",
66
"0.4": "0.4.3",
7-
"0.5": "0.5.2"
7+
"0.5": "0.5.2",
8+
"0.6": "0.6.1"
89
},
910
"versions": {
1011
"0.3.1": {
@@ -101,6 +102,7 @@
101102
"us-west-2": "763104351884"
102103
},
103104
"repository": "autogluon-training",
105+
"processors": ["cpu", "gpu"],
104106
"py_versions": ["py38"]
105107
},
106108
"0.4.2": {
@@ -133,6 +135,7 @@
133135
"us-west-2": "763104351884"
134136
},
135137
"repository": "autogluon-training",
138+
"processors": ["cpu", "gpu"],
136139
"py_versions": ["py38"]
137140
},
138141
"0.4.3": {
@@ -165,6 +168,7 @@
165168
"us-west-2": "763104351884"
166169
},
167170
"repository": "autogluon-training",
171+
"processors": ["cpu", "gpu"],
168172
"py_versions": ["py38"]
169173
},
170174
"0.5.2": {
@@ -197,6 +201,39 @@
197201
"us-west-2": "763104351884"
198202
},
199203
"repository": "autogluon-training",
204+
"processors": ["cpu", "gpu"],
205+
"py_versions": ["py38"]
206+
},
207+
"0.6.1": {
208+
"registries": {
209+
"af-south-1": "626614931356",
210+
"ap-east-1": "871362719292",
211+
"ap-northeast-1": "763104351884",
212+
"ap-northeast-2": "763104351884",
213+
"ap-northeast-3": "364406365360",
214+
"ap-south-1": "763104351884",
215+
"ap-southeast-1": "763104351884",
216+
"ap-southeast-2": "763104351884",
217+
"ap-southeast-3": "907027046896",
218+
"ca-central-1": "763104351884",
219+
"eu-central-1": "763104351884",
220+
"eu-north-1": "763104351884",
221+
"eu-west-1": "763104351884",
222+
"eu-west-2": "763104351884",
223+
"eu-west-3": "763104351884",
224+
"eu-south-1": "692866216735",
225+
"me-south-1": "217643126080",
226+
"sa-east-1": "763104351884",
227+
"us-east-1": "763104351884",
228+
"us-east-2": "763104351884",
229+
"us-gov-east-1": "446045086412",
230+
"us-gov-west-1": "442386744353",
231+
"us-iso-east-1": "886529160074",
232+
"us-west-1": "763104351884",
233+
"us-west-2": "763104351884"
234+
},
235+
"repository": "autogluon-training",
236+
"processors": ["cpu", "gpu"],
200237
"py_versions": ["py38"]
201238
}
202239
}
@@ -205,7 +242,8 @@
205242
"version_aliases": {
206243
"0.3": "0.3.2",
207244
"0.4": "0.4.3",
208-
"0.5": "0.5.2"
245+
"0.5": "0.5.2",
246+
"0.6": "0.6.1"
209247
},
210248
"versions": {
211249
"0.3.1": {
@@ -435,6 +473,40 @@
435473
"repository": "autogluon-inference",
436474
"processors": ["cpu", "gpu"],
437475
"py_versions": ["py38"]
476+
},
477+
"0.6.1": {
478+
"registries": {
479+
"af-south-1": "626614931356",
480+
"ap-east-1": "871362719292",
481+
"ap-northeast-1": "763104351884",
482+
"ap-northeast-2": "763104351884",
483+
"ap-northeast-3": "364406365360",
484+
"ap-south-1": "763104351884",
485+
"ap-southeast-1": "763104351884",
486+
"ap-southeast-2": "763104351884",
487+
"ap-southeast-3": "907027046896",
488+
"ca-central-1": "763104351884",
489+
"cn-north-1": "727897471807",
490+
"cn-northwest-1": "727897471807",
491+
"eu-central-1": "763104351884",
492+
"eu-north-1": "763104351884",
493+
"eu-west-1": "763104351884",
494+
"eu-west-2": "763104351884",
495+
"eu-west-3": "763104351884",
496+
"eu-south-1": "692866216735",
497+
"me-south-1": "217643126080",
498+
"sa-east-1": "763104351884",
499+
"us-east-1": "763104351884",
500+
"us-east-2": "763104351884",
501+
"us-gov-east-1": "446045086412",
502+
"us-gov-west-1": "442386744353",
503+
"us-iso-east-1": "886529160074",
504+
"us-west-1": "763104351884",
505+
"us-west-2": "763104351884"
506+
},
507+
"repository": "autogluon-inference",
508+
"processors": ["cpu", "gpu"],
509+
"py_versions": ["py38"]
438510
}
439511
}
440512
}

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

+14-1
Original file line numberDiff line numberDiff line change
@@ -37,12 +37,25 @@
3737
"sa-east-1": "763104351884",
3838
"us-east-1": "763104351884",
3939
"us-east-2": "763104351884",
40+
"us-gov-east-1": "446045086412",
4041
"us-gov-west-1": "442386744353",
4142
"us-iso-east-1": "886529160074",
4243
"us-west-1": "763104351884",
4344
"us-west-2": "763104351884",
4445
}
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"]
46+
VERSIONS = [
47+
"0.3.1",
48+
"0.3.2",
49+
"0.4.0",
50+
"0.4.2",
51+
"0.4.3",
52+
"0.3",
53+
"0.4",
54+
"0.5.2",
55+
"0.5",
56+
"0.6.1",
57+
"0.6",
58+
]
4659
SCOPES = ["training", "inference"]
4760
PROCESSORS = ["cpu", "gpu"]
4861

0 commit comments

Comments
 (0)