Skip to content

Commit 5a877f8

Browse files
NikhilRaverkarNikhil Raverkar
authored and
Namrata Madan
committed
feature: Add XGBoost framework 1.7-1 version (aws#3706)
Co-authored-by: Nikhil Raverkar <[email protected]>
1 parent b5462d9 commit 5a877f8

File tree

2 files changed

+69
-1
lines changed

2 files changed

+69
-1
lines changed

src/sagemaker/image_uri_config/xgboost.json

+68
Original file line numberDiff line numberDiff line change
@@ -281,6 +281,40 @@
281281
"us-west-2": "246618743249"
282282
},
283283
"repository": "sagemaker-xgboost"
284+
},
285+
"1.7-1": {
286+
"registries": {
287+
"af-south-1": "510948584623",
288+
"ap-east-1": "651117190479",
289+
"ap-northeast-1": "354813040037",
290+
"ap-northeast-2": "366743142698",
291+
"ap-northeast-3": "867004704886",
292+
"ap-south-1": "720646828776",
293+
"ap-southeast-1": "121021644041",
294+
"ap-southeast-2": "783357654285",
295+
"ap-southeast-3": "951798379941",
296+
"ca-central-1": "341280168497",
297+
"cn-north-1": "450853457545",
298+
"cn-northwest-1": "451049120500",
299+
"eu-central-1": "492215442770",
300+
"eu-north-1": "662702820516",
301+
"eu-west-1": "141502667606",
302+
"eu-west-2": "764974769150",
303+
"eu-west-3": "659782779980",
304+
"eu-south-1": "978288397137",
305+
"me-south-1": "801668240914",
306+
"me-central-1": "272398656194",
307+
"sa-east-1": "737474898029",
308+
"us-east-1": "683313688378",
309+
"us-east-2": "257758044811",
310+
"us-gov-west-1": "414596584902",
311+
"us-gov-east-1": "237065988967",
312+
"us-iso-east-1": "833128469047",
313+
"us-isob-east-1": "281123927165",
314+
"us-west-1": "746614075791",
315+
"us-west-2": "246618743249"
316+
},
317+
"repository": "sagemaker-xgboost"
284318
}
285319
}
286320
},
@@ -566,6 +600,40 @@
566600
"us-west-2": "246618743249"
567601
},
568602
"repository": "sagemaker-xgboost"
603+
},
604+
"1.7-1": {
605+
"registries": {
606+
"af-south-1": "510948584623",
607+
"ap-east-1": "651117190479",
608+
"ap-northeast-1": "354813040037",
609+
"ap-northeast-2": "366743142698",
610+
"ap-northeast-3": "867004704886",
611+
"ap-south-1": "720646828776",
612+
"ap-southeast-1": "121021644041",
613+
"ap-southeast-2": "783357654285",
614+
"ap-southeast-3": "951798379941",
615+
"ca-central-1": "341280168497",
616+
"cn-north-1": "450853457545",
617+
"cn-northwest-1": "451049120500",
618+
"eu-central-1": "492215442770",
619+
"eu-north-1": "662702820516",
620+
"eu-west-1": "141502667606",
621+
"eu-west-2": "764974769150",
622+
"eu-west-3": "659782779980",
623+
"eu-south-1": "978288397137",
624+
"me-south-1": "801668240914",
625+
"me-central-1": "272398656194",
626+
"sa-east-1": "737474898029",
627+
"us-east-1": "683313688378",
628+
"us-east-2": "257758044811",
629+
"us-gov-west-1": "414596584902",
630+
"us-gov-east-1": "237065988967",
631+
"us-iso-east-1": "833128469047",
632+
"us-isob-east-1": "281123927165",
633+
"us-west-1": "746614075791",
634+
"us-west-2": "246618743249"
635+
},
636+
"repository": "sagemaker-xgboost"
569637
}
570638
}
571639
},

tests/unit/sagemaker/image_uris/test_xgboost.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@
4848
}
4949
ALGO_VERSIONS = ("1", "latest")
5050
XGBOOST_FRAMEWORK_CPU_ONLY_VERSIONS = ("0.90-2", "0.90-1", "1.0-1")
51-
XGBOOST_FRAMEWORK_CPU_GPU_VERSIONS = ("1.2-1", "1.2-2", "1.3-1", "1.5-1")
51+
XGBOOST_FRAMEWORK_CPU_GPU_VERSIONS = ("1.2-1", "1.2-2", "1.3-1", "1.5-1", "1.7-1")
5252

5353
FRAMEWORK_REGISTRIES = {
5454
"af-south-1": "510948584623",

0 commit comments

Comments
 (0)