Skip to content

Commit 0502274

Browse files
authored
feature: add xgboost framework version 1.5-1 (aws#3044)
1 parent c96516c commit 0502274

File tree

2 files changed

+31
-1
lines changed

2 files changed

+31
-1
lines changed

src/sagemaker/image_uri_config/xgboost.json

+30
Original file line numberDiff line numberDiff line change
@@ -219,6 +219,36 @@
219219
"us-west-2": "246618743249"
220220
},
221221
"repository": "sagemaker-xgboost"
222+
},
223+
"1.5-1": {
224+
"registries": {
225+
"af-south-1": "510948584623",
226+
"ap-east-1": "651117190479",
227+
"ap-northeast-1": "354813040037",
228+
"ap-northeast-2": "366743142698",
229+
"ap-northeast-3": "867004704886",
230+
"ap-south-1": "720646828776",
231+
"ap-southeast-1": "121021644041",
232+
"ap-southeast-2": "783357654285",
233+
"ca-central-1": "341280168497",
234+
"cn-north-1": "450853457545",
235+
"cn-northwest-1": "451049120500",
236+
"eu-central-1": "492215442770",
237+
"eu-north-1": "662702820516",
238+
"eu-west-1": "141502667606",
239+
"eu-west-2": "764974769150",
240+
"eu-west-3": "659782779980",
241+
"eu-south-1": "978288397137",
242+
"me-south-1": "801668240914",
243+
"sa-east-1": "737474898029",
244+
"us-east-1": "683313688378",
245+
"us-east-2": "257758044811",
246+
"us-gov-west-1": "414596584902",
247+
"us-iso-east-1": "833128469047",
248+
"us-west-1": "746614075791",
249+
"us-west-2": "246618743249"
250+
},
251+
"repository": "sagemaker-xgboost"
222252
}
223253
}
224254
}

tests/unit/sagemaker/image_uris/test_xgboost.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@
4444
}
4545
ALGO_VERSIONS = ("1", "latest")
4646
XGBOOST_FRAMEWORK_CPU_ONLY_VERSIONS = ("0.90-2", "0.90-1", "1.0-1")
47-
XGBOOST_FRAMEWORK_CPU_GPU_VERSIONS = ("1.2-1", "1.2-2", "1.3-1")
47+
XGBOOST_FRAMEWORK_CPU_GPU_VERSIONS = ("1.2-1", "1.2-2", "1.3-1", "1.5-1")
4848

4949
FRAMEWORK_REGISTRIES = {
5050
"af-south-1": "510948584623",

0 commit comments

Comments
 (0)