Skip to content

Commit ae1a67c

Browse files
author
Roja Reddy Sareddy
committed
update SKLearn image URI config
1 parent 9884e7b commit ae1a67c

File tree

2 files changed

+85
-77
lines changed

2 files changed

+85
-77
lines changed

‎src/sagemaker/image_uri_config/sklearn.json‎

Lines changed: 84 additions & 76 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,8 @@
22
"inference": {
33
"versions": {
44
"0.20.0": {
5-
"processors": [
6-
"cpu"
7-
],
8-
"py_versions": [
9-
"py3"
10-
],
5+
"processors": ["cpu"],
6+
"py_versions": ["py3"],
117
"registries": {
128
"af-south-1": "510948584623",
139
"ap-east-1": "651117190479",
@@ -45,17 +41,13 @@
4541
"us-isof-east-1": "108575199400",
4642
"us-isof-south-1": "124985052026",
4743
"us-west-1": "746614075791",
48-
"us-west-2": "246618743249"
44+
"us-west-2": "246618743249",
4945
},
50-
"repository": "sagemaker-scikit-learn"
46+
"repository": "sagemaker-scikit-learn",
5147
},
5248
"0.23-1": {
53-
"processors": [
54-
"cpu"
55-
],
56-
"py_versions": [
57-
"py3"
58-
],
49+
"processors": ["cpu"],
50+
"py_versions": ["py3"],
5951
"registries": {
6052
"af-south-1": "510948584623",
6153
"ap-east-1": "651117190479",
@@ -93,17 +85,13 @@
9385
"us-isof-east-1": "108575199400",
9486
"us-isof-south-1": "124985052026",
9587
"us-west-1": "746614075791",
96-
"us-west-2": "246618743249"
88+
"us-west-2": "246618743249",
9789
},
98-
"repository": "sagemaker-scikit-learn"
90+
"repository": "sagemaker-scikit-learn",
9991
},
10092
"1.0-1": {
101-
"processors": [
102-
"cpu"
103-
],
104-
"py_versions": [
105-
"py3"
106-
],
93+
"processors": ["cpu"],
94+
"py_versions": ["py3"],
10795
"registries": {
10896
"af-south-1": "510948584623",
10997
"ap-east-1": "651117190479",
@@ -141,17 +129,13 @@
141129
"us-isof-east-1": "108575199400",
142130
"us-isof-south-1": "124985052026",
143131
"us-west-1": "746614075791",
144-
"us-west-2": "246618743249"
132+
"us-west-2": "246618743249",
145133
},
146-
"repository": "sagemaker-scikit-learn"
134+
"repository": "sagemaker-scikit-learn",
147135
},
148136
"1.2-1": {
149-
"processors": [
150-
"cpu"
151-
],
152-
"py_versions": [
153-
"py3"
154-
],
137+
"processors": ["cpu"],
138+
"py_versions": ["py3"],
155139
"registries": {
156140
"af-south-1": "510948584623",
157141
"ap-east-1": "651117190479",
@@ -189,21 +173,17 @@
189173
"us-isof-east-1": "108575199400",
190174
"us-isof-south-1": "124985052026",
191175
"us-west-1": "746614075791",
192-
"us-west-2": "246618743249"
176+
"us-west-2": "246618743249",
193177
},
194-
"repository": "sagemaker-scikit-learn"
195-
}
178+
"repository": "sagemaker-scikit-learn",
179+
},
196180
}
197181
},
198182
"training": {
199183
"versions": {
200184
"0.20.0": {
201-
"processors": [
202-
"cpu"
203-
],
204-
"py_versions": [
205-
"py3"
206-
],
185+
"processors": ["cpu"],
186+
"py_versions": ["py3"],
207187
"registries": {
208188
"af-south-1": "510948584623",
209189
"ap-east-1": "651117190479",
@@ -241,17 +221,13 @@
241221
"us-isof-east-1": "108575199400",
242222
"us-isof-south-1": "124985052026",
243223
"us-west-1": "746614075791",
244-
"us-west-2": "246618743249"
224+
"us-west-2": "246618743249",
245225
},
246-
"repository": "sagemaker-scikit-learn"
226+
"repository": "sagemaker-scikit-learn",
247227
},
248228
"0.23-1": {
249-
"processors": [
250-
"cpu"
251-
],
252-
"py_versions": [
253-
"py3"
254-
],
229+
"processors": ["cpu"],
230+
"py_versions": ["py3"],
255231
"registries": {
256232
"af-south-1": "510948584623",
257233
"ap-east-1": "651117190479",
@@ -289,17 +265,13 @@
289265
"us-isof-east-1": "108575199400",
290266
"us-isof-south-1": "124985052026",
291267
"us-west-1": "746614075791",
292-
"us-west-2": "246618743249"
268+
"us-west-2": "246618743249",
293269
},
294-
"repository": "sagemaker-scikit-learn"
270+
"repository": "sagemaker-scikit-learn",
295271
},
296272
"1.0-1": {
297-
"processors": [
298-
"cpu"
299-
],
300-
"py_versions": [
301-
"py3"
302-
],
273+
"processors": ["cpu"],
274+
"py_versions": ["py3"],
303275
"registries": {
304276
"af-south-1": "510948584623",
305277
"ap-east-1": "651117190479",
@@ -337,17 +309,13 @@
337309
"us-isof-east-1": "108575199400",
338310
"us-isof-south-1": "124985052026",
339311
"us-west-1": "746614075791",
340-
"us-west-2": "246618743249"
312+
"us-west-2": "246618743249",
341313
},
342-
"repository": "sagemaker-scikit-learn"
314+
"repository": "sagemaker-scikit-learn",
343315
},
344316
"1.2-1": {
345-
"processors": [
346-
"cpu"
347-
],
348-
"py_versions": [
349-
"py3"
350-
],
317+
"processors": ["cpu"],
318+
"py_versions": ["py3"],
351319
"registries": {
352320
"af-south-1": "510948584623",
353321
"ap-east-1": "651117190479",
@@ -385,21 +353,61 @@
385353
"us-isof-east-1": "108575199400",
386354
"us-isof-south-1": "124985052026",
387355
"us-west-1": "746614075791",
388-
"us-west-2": "246618743249"
356+
"us-west-2": "246618743249",
389357
},
390-
"repository": "sagemaker-scikit-learn"
391-
}
358+
"repository": "sagemaker-scikit-learn",
359+
},
360+
"1.4-2": {
361+
"processors": ["cpu"],
362+
"py_versions": ["py3"],
363+
"registries": {
364+
"af-south-1": "510948584623",
365+
"ap-east-1": "651117190479",
366+
"ap-northeast-1": "354813040037",
367+
"ap-northeast-2": "366743142698",
368+
"ap-northeast-3": "867004704886",
369+
"ap-south-1": "720646828776",
370+
"ap-south-2": "628508329040",
371+
"ap-southeast-1": "121021644041",
372+
"ap-southeast-2": "783357654285",
373+
"ap-southeast-3": "951798379941",
374+
"ap-southeast-4": "106583098589",
375+
"ca-central-1": "341280168497",
376+
"ca-west-1": "190319476487",
377+
"cn-north-1": "450853457545",
378+
"cn-northwest-1": "451049120500",
379+
"eu-central-1": "492215442770",
380+
"eu-central-2": "680994064768",
381+
"eu-north-1": "662702820516",
382+
"eu-south-1": "978288397137",
383+
"eu-south-2": "104374241257",
384+
"eu-west-1": "141502667606",
385+
"eu-west-2": "764974769150",
386+
"eu-west-3": "659782779980",
387+
"il-central-1": "898809789911",
388+
"me-central-1": "272398656194",
389+
"me-south-1": "801668240914",
390+
"sa-east-1": "737474898029",
391+
"us-east-1": "683313688378",
392+
"us-east-2": "257758044811",
393+
"us-gov-east-1": "237065988967",
394+
"us-gov-west-1": "414596584902",
395+
"us-iso-east-1": "833128469047",
396+
"us-isob-east-1": "281123927165",
397+
"us-isof-east-1": "108575199400",
398+
"us-isof-south-1": "124985052026",
399+
"us-west-1": "746614075791",
400+
"us-west-2": "246618743249",
401+
},
402+
"repository": "sagemaker-scikit-learn",
403+
},
392404
}
393405
},
394406
"inference_graviton": {
395407
"versions": {
396408
"1.0-1": {
397-
"processors": [
398-
"cpu"
399-
],
400-
"py_versions": [
401-
"py3"
402-
],
409+
"processors": ["cpu"],
410+
"py_versions": ["py3"],
403411
"registries": {
404412
"af-south-1": "510948584623",
405413
"ap-east-1": "651117190479",
@@ -437,10 +445,10 @@
437445
"us-isof-east-1": "108575199400",
438446
"us-isof-south-1": "124985052026",
439447
"us-west-1": "746614075791",
440-
"us-west-2": "246618743249"
448+
"us-west-2": "246618743249",
441449
},
442-
"repository": "sagemaker-scikit-learn"
450+
"repository": "sagemaker-scikit-learn",
443451
}
444452
}
445-
}
446-
}
453+
},
454+
}

‎tests/integ/sagemaker/experiments/test_run.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -693,7 +693,7 @@ def _generate_estimator(
693693
sagemaker_client_config=sagemaker_client_config,
694694
)
695695
return SKLearn(
696-
framework_version="1.2-1",
696+
framework_version="1.4-2",
697697
entry_point=_ENTRY_POINT_PATH,
698698
dependencies=[sdk_tar],
699699
role=execution_role,

0 commit comments

Comments
 (0)