Skip to content

Commit e6e8649

Browse files
lxninglxning
committed
update sklearn from 1.2-1 to 1.2-1-1
1 parent fda438c commit e6e8649

File tree

6 files changed

+105
-10
lines changed

6 files changed

+105
-10
lines changed

‎src/sagemaker/image_uri_config/sklearn.json‎

Lines changed: 96 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -192,6 +192,54 @@
192192
"us-west-2": "246618743249"
193193
},
194194
"repository": "sagemaker-scikit-learn"
195+
},
196+
"1.2-1-1": {
197+
"processors": [
198+
"cpu"
199+
],
200+
"py_versions": [
201+
"py3"
202+
],
203+
"registries": {
204+
"af-south-1": "510948584623",
205+
"ap-east-1": "651117190479",
206+
"ap-northeast-1": "354813040037",
207+
"ap-northeast-2": "366743142698",
208+
"ap-northeast-3": "867004704886",
209+
"ap-south-1": "720646828776",
210+
"ap-south-2": "628508329040",
211+
"ap-southeast-1": "121021644041",
212+
"ap-southeast-2": "783357654285",
213+
"ap-southeast-3": "951798379941",
214+
"ap-southeast-4": "106583098589",
215+
"ca-central-1": "341280168497",
216+
"ca-west-1": "190319476487",
217+
"cn-north-1": "450853457545",
218+
"cn-northwest-1": "451049120500",
219+
"eu-central-1": "492215442770",
220+
"eu-central-2": "680994064768",
221+
"eu-north-1": "662702820516",
222+
"eu-south-1": "978288397137",
223+
"eu-south-2": "104374241257",
224+
"eu-west-1": "141502667606",
225+
"eu-west-2": "764974769150",
226+
"eu-west-3": "659782779980",
227+
"il-central-1": "898809789911",
228+
"me-central-1": "272398656194",
229+
"me-south-1": "801668240914",
230+
"sa-east-1": "737474898029",
231+
"us-east-1": "683313688378",
232+
"us-east-2": "257758044811",
233+
"us-gov-east-1": "237065988967",
234+
"us-gov-west-1": "414596584902",
235+
"us-iso-east-1": "833128469047",
236+
"us-isob-east-1": "281123927165",
237+
"us-isof-east-1": "108575199400",
238+
"us-isof-south-1": "124985052026",
239+
"us-west-1": "746614075791",
240+
"us-west-2": "246618743249"
241+
},
242+
"repository": "sagemaker-scikit-learn"
195243
}
196244
}
197245
},
@@ -388,6 +436,54 @@
388436
"us-west-2": "246618743249"
389437
},
390438
"repository": "sagemaker-scikit-learn"
439+
},
440+
"1.2-1-1": {
441+
"processors": [
442+
"cpu"
443+
],
444+
"py_versions": [
445+
"py3"
446+
],
447+
"registries": {
448+
"af-south-1": "510948584623",
449+
"ap-east-1": "651117190479",
450+
"ap-northeast-1": "354813040037",
451+
"ap-northeast-2": "366743142698",
452+
"ap-northeast-3": "867004704886",
453+
"ap-south-1": "720646828776",
454+
"ap-south-2": "628508329040",
455+
"ap-southeast-1": "121021644041",
456+
"ap-southeast-2": "783357654285",
457+
"ap-southeast-3": "951798379941",
458+
"ap-southeast-4": "106583098589",
459+
"ca-central-1": "341280168497",
460+
"ca-west-1": "190319476487",
461+
"cn-north-1": "450853457545",
462+
"cn-northwest-1": "451049120500",
463+
"eu-central-1": "492215442770",
464+
"eu-central-2": "680994064768",
465+
"eu-north-1": "662702820516",
466+
"eu-south-1": "978288397137",
467+
"eu-south-2": "104374241257",
468+
"eu-west-1": "141502667606",
469+
"eu-west-2": "764974769150",
470+
"eu-west-3": "659782779980",
471+
"il-central-1": "898809789911",
472+
"me-central-1": "272398656194",
473+
"me-south-1": "801668240914",
474+
"sa-east-1": "737474898029",
475+
"us-east-1": "683313688378",
476+
"us-east-2": "257758044811",
477+
"us-gov-east-1": "237065988967",
478+
"us-gov-west-1": "414596584902",
479+
"us-iso-east-1": "833128469047",
480+
"us-isob-east-1": "281123927165",
481+
"us-isof-east-1": "108575199400",
482+
"us-isof-south-1": "124985052026",
483+
"us-west-1": "746614075791",
484+
"us-west-2": "246618743249"
485+
},
486+
"repository": "sagemaker-scikit-learn"
391487
}
392488
}
393489
},

‎src/sagemaker/workflow/_utils.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@
4646

4747
logger = logging.getLogger(__name__)
4848

49-
FRAMEWORK_VERSION = "1.2-1"
49+
FRAMEWORK_VERSION = "1.2-1-1"
5050
INSTANCE_TYPE = "ml.m5.large"
5151
REPACK_SCRIPT = "_repack_model.py"
5252
REPACK_SCRIPT_LAUNCHER = "_repack_script_launcher.sh"

‎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.2-1-1",
697697
entry_point=_ENTRY_POINT_PATH,
698698
dependencies=[sdk_tar],
699699
role=execution_role,

‎tests/unit/sagemaker/image_uris/jumpstart/test_sklearn.py‎

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -56,13 +56,12 @@ def test_jumpstart_sklearn_image_uri(patched_get_model_specs, session):
5656
# framework classes dont use digest.
5757
assert (
5858
framework_class_uri
59-
== "246618743249.dkr.ecr.us-west-2.amazonaws.com/sagemaker-scikit-learn:1.2-1"
59+
== "246618743249.dkr.ecr.us-west-2.amazonaws.com/sagemaker-scikit-learn:1.2-1-1"
6060
"-cpu-py3"
6161
)
6262
assert (
6363
uri == "246618743249.dkr.ecr.us-west-2.amazonaws.com/sagemaker-scikit-learn@"
64-
"sha256:e09bbb7686077a1db23d316b699020a786a6e1636b2b89384be9651368c40f95"
65-
)
64+
"sha256:a7a5a9097f078d3d0ccb7248ec076ad63e10c82e39a798408f5066450afe82bd"
6665

6766
# training
6867
uri = image_uris.retrieve(
@@ -88,11 +87,11 @@ def test_jumpstart_sklearn_image_uri(patched_get_model_specs, session):
8887
# framework classes dont use digest.
8988
assert (
9089
framework_class_uri
91-
== "246618743249.dkr.ecr.us-west-2.amazonaws.com/sagemaker-scikit-learn:1.2-1-cpu-py3"
90+
== "246618743249.dkr.ecr.us-west-2.amazonaws.com/sagemaker-scikit-learn:1.2-1-1-cpu-py3"
9291
)
9392
assert (
9493
uri == "246618743249.dkr.ecr.us-west-2.amazonaws.com/sagemaker-scikit-learn"
95-
"@sha256:e09bbb7686077a1db23d316b699020a786a6e1636b2b89384be9651368c40f95"
94+
"@sha256:a7a5a9097f078d3d0ccb7248ec076ad63e10c82e39a798408f5066450afe82bd"
9695
)
9796

9897
with pytest.raises(ValueError):

‎tests/unit/sagemaker/jumpstart/constants.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14854,7 +14854,7 @@
1485414854
"incremental_training_supported": False,
1485514855
"hosting_ecr_specs": {
1485614856
"framework": "sklearn",
14857-
"framework_version": "1.2-1",
14857+
"framework_version": "1.2-1-1",
1485814858
"py_version": "py3",
1485914859
},
1486014860
"hosting_artifact_key": "sklearn-infer/infer-sklearn-classification-linear.tar.gz",
@@ -14974,7 +14974,7 @@
1497414974
"training_script_key": "source-directory-tarballs/training/sklearn-classification/v2.0.1/sourcedir.tar.gz",
1497514975
"training_ecr_specs": {
1497614976
"framework": "sklearn",
14977-
"framework_version": "1.2-1",
14977+
"framework_version": "1.2-1-1",
1497814978
"py_version": "py3",
1497914979
},
1498014980
"training_artifact_key": "sklearn-training/train-sklearn-classification-linear.tar.gz",

‎tests/unit/sagemaker/workflow/test_step_collections.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@
5555

5656
MODEL_NAME = "gisele"
5757
MODEL_REPACKING_IMAGE_URI = (
58-
"246618743249.dkr.ecr.us-west-2.amazonaws.com/sagemaker-scikit-learn:1.2-1-cpu-py3"
58+
"246618743249.dkr.ecr.us-west-2.amazonaws.com/sagemaker-scikit-learn:1.2-1-1-cpu-py3"
5959
)
6060

6161

0 commit comments

Comments
 (0)