Skip to content

Commit fb50e76

Browse files
gradientskygradientsky
authored
feature: AutoGluon 0.3.2 and 0.4.0 image_uris (#2991)
1 parent a065e8b commit fb50e76

File tree

2 files changed

+132
-3
lines changed

2 files changed

+132
-3
lines changed

‎src/sagemaker/image_uri_config/autogluon.json

+128
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,10 @@
11
{
22
"training": {
33
"processors": ["cpu", "gpu"],
4+
"version_aliases": {
5+
"0.3": "0.3.2",
6+
"0.4": "0.4.0"
7+
},
48
"versions": {
59
"0.3.1": {
610
"registries": {
@@ -30,11 +34,73 @@
3034
},
3135
"repository": "autogluon-training",
3236
"py_versions": ["py37"]
37+
},
38+
"0.3.2": {
39+
"registries": {
40+
"af-south-1": "626614931356",
41+
"ap-east-1": "871362719292",
42+
"ap-northeast-1": "763104351884",
43+
"ap-northeast-2": "763104351884",
44+
"ap-northeast-3": "364406365360",
45+
"ap-south-1": "763104351884",
46+
"ap-southeast-1": "763104351884",
47+
"ap-southeast-2": "763104351884",
48+
"ca-central-1": "763104351884",
49+
"eu-central-1": "763104351884",
50+
"eu-north-1": "763104351884",
51+
"eu-west-1": "763104351884",
52+
"eu-west-2": "763104351884",
53+
"eu-west-3": "763104351884",
54+
"eu-south-1": "692866216735",
55+
"me-south-1": "217643126080",
56+
"sa-east-1": "763104351884",
57+
"us-east-1": "763104351884",
58+
"us-east-2": "763104351884",
59+
"us-gov-west-1": "442386744353",
60+
"us-iso-east-1": "886529160074",
61+
"us-west-1": "763104351884",
62+
"us-west-2": "763104351884"
63+
},
64+
"repository": "autogluon-training",
65+
"py_versions": ["py38"]
66+
},
67+
"0.4.0": {
68+
"registries": {
69+
"af-south-1": "626614931356",
70+
"ap-east-1": "871362719292",
71+
"ap-northeast-1": "763104351884",
72+
"ap-northeast-2": "763104351884",
73+
"ap-northeast-3": "364406365360",
74+
"ap-south-1": "763104351884",
75+
"ap-southeast-1": "763104351884",
76+
"ap-southeast-2": "763104351884",
77+
"ca-central-1": "763104351884",
78+
"eu-central-1": "763104351884",
79+
"eu-north-1": "763104351884",
80+
"eu-west-1": "763104351884",
81+
"eu-west-2": "763104351884",
82+
"eu-west-3": "763104351884",
83+
"eu-south-1": "692866216735",
84+
"me-south-1": "217643126080",
85+
"sa-east-1": "763104351884",
86+
"us-east-1": "763104351884",
87+
"us-east-2": "763104351884",
88+
"us-gov-west-1": "442386744353",
89+
"us-iso-east-1": "886529160074",
90+
"us-west-1": "763104351884",
91+
"us-west-2": "763104351884"
92+
},
93+
"repository": "autogluon-training",
94+
"py_versions": ["py38"]
3395
}
3496
}
3597
},
3698
"inference": {
3799
"processors": ["cpu"],
100+
"version_aliases": {
101+
"0.3": "0.3.2",
102+
"0.4": "0.4.0"
103+
},
38104
"versions": {
39105
"0.3.1": {
40106
"registries": {
@@ -66,6 +132,68 @@
66132
},
67133
"repository": "autogluon-inference",
68134
"py_versions": ["py37"]
135+
},
136+
"0.3.2": {
137+
"registries": {
138+
"af-south-1": "626614931356",
139+
"ap-east-1": "871362719292",
140+
"ap-northeast-1": "763104351884",
141+
"ap-northeast-2": "763104351884",
142+
"ap-northeast-3": "364406365360",
143+
"ap-south-1": "763104351884",
144+
"ap-southeast-1": "763104351884",
145+
"ap-southeast-2": "763104351884",
146+
"ca-central-1": "763104351884",
147+
"cn-north-1": "727897471807",
148+
"cn-northwest-1": "727897471807",
149+
"eu-central-1": "763104351884",
150+
"eu-north-1": "763104351884",
151+
"eu-west-1": "763104351884",
152+
"eu-west-2": "763104351884",
153+
"eu-west-3": "763104351884",
154+
"eu-south-1": "692866216735",
155+
"me-south-1": "217643126080",
156+
"sa-east-1": "763104351884",
157+
"us-east-1": "763104351884",
158+
"us-east-2": "763104351884",
159+
"us-gov-west-1": "442386744353",
160+
"us-iso-east-1": "886529160074",
161+
"us-west-1": "763104351884",
162+
"us-west-2": "763104351884"
163+
},
164+
"repository": "autogluon-inference",
165+
"py_versions": ["py38"]
166+
},
167+
"0.4.0": {
168+
"registries": {
169+
"af-south-1": "626614931356",
170+
"ap-east-1": "871362719292",
171+
"ap-northeast-1": "763104351884",
172+
"ap-northeast-2": "763104351884",
173+
"ap-northeast-3": "364406365360",
174+
"ap-south-1": "763104351884",
175+
"ap-southeast-1": "763104351884",
176+
"ap-southeast-2": "763104351884",
177+
"ca-central-1": "763104351884",
178+
"cn-north-1": "727897471807",
179+
"cn-northwest-1": "727897471807",
180+
"eu-central-1": "763104351884",
181+
"eu-north-1": "763104351884",
182+
"eu-west-1": "763104351884",
183+
"eu-west-2": "763104351884",
184+
"eu-west-3": "763104351884",
185+
"eu-south-1": "692866216735",
186+
"me-south-1": "217643126080",
187+
"sa-east-1": "763104351884",
188+
"us-east-1": "763104351884",
189+
"us-east-2": "763104351884",
190+
"us-gov-west-1": "442386744353",
191+
"us-iso-east-1": "886529160074",
192+
"us-west-1": "763104351884",
193+
"us-west-2": "763104351884"
194+
},
195+
"repository": "autogluon-inference",
196+
"py_versions": ["py38"]
69197
}
70198
}
71199
}

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

+4-3
Original file line numberDiff line numberDiff line change
@@ -42,17 +42,18 @@
4242
"us-west-1": "763104351884",
4343
"us-west-2": "763104351884",
4444
}
45-
VERSIONS = ["0.3.1"]
45+
VERSIONS = ["0.3.1", "0.3.2", "0.4.0", "0.3", "0.4"]
4646

4747

4848
@pytest.mark.parametrize("version", VERSIONS)
4949
def test_valid_uris(version):
50+
py_version = "py37" if version == "0.3.1" else "py38"
5051
for region in ACCOUNTS.keys():
5152
uri = image_uris.retrieve(
5253
"autogluon",
5354
region=region,
5455
version=version,
55-
py_version="py37",
56+
py_version=py_version,
5657
image_scope="training",
5758
instance_type="ml.c4.xlarge",
5859
)
@@ -61,7 +62,7 @@ def test_valid_uris(version):
6162
"autogluon-training",
6263
version,
6364
ACCOUNTS[region],
64-
py_version="py37",
65+
py_version=py_version,
6566
region=region,
6667
)
6768
assert uri == expected

0 commit comments

Comments
 (0)