Skip to content

Commit 5333974

Browse files
Edward J Kimicywang86rui
Edward J Kim
andauthored
feature: add xgboost framework version 1.2-2 (#2324)
Co-authored-by: icywang86rui <[email protected]>
1 parent 529752c commit 5333974

File tree

2 files changed

+30
-1
lines changed

2 files changed

+30
-1
lines changed

src/sagemaker/image_uri_config/xgboost.json

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,35 @@
154154
"us-west-2": "246618743249"
155155
},
156156
"repository": "sagemaker-xgboost"
157+
},
158+
"1.2-2": {
159+
"registries": {
160+
"af-south-1": "510948584623",
161+
"ap-east-1": "651117190479",
162+
"ap-northeast-1": "354813040037",
163+
"ap-northeast-2": "366743142698",
164+
"ap-south-1": "720646828776",
165+
"ap-southeast-1": "121021644041",
166+
"ap-southeast-2": "783357654285",
167+
"ca-central-1": "341280168497",
168+
"cn-north-1": "450853457545",
169+
"cn-northwest-1": "451049120500",
170+
"eu-central-1": "492215442770",
171+
"eu-north-1": "662702820516",
172+
"eu-west-1": "141502667606",
173+
"eu-west-2": "764974769150",
174+
"eu-west-3": "659782779980",
175+
"eu-south-1": "978288397137",
176+
"me-south-1": "801668240914",
177+
"sa-east-1": "737474898029",
178+
"us-east-1": "683313688378",
179+
"us-east-2": "257758044811",
180+
"us-gov-west-1": "414596584902",
181+
"us-iso-east-1": "833128469047",
182+
"us-west-1": "746614075791",
183+
"us-west-2": "246618743249"
184+
},
185+
"repository": "sagemaker-xgboost"
157186
}
158187
}
159188
}

tests/unit/sagemaker/image_uris/test_xgboost.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@
4545
}
4646
ALGO_VERSIONS = ("1", "latest")
4747
XGBOOST_FRAMEWORK_CPU_ONLY_VERSIONS = ("0.90-2", "0.90-1", "1.0-1")
48-
XGBOOST_FRAMEWORK_CPU_GPU_VERSIONS = ("1.2-1",)
48+
XGBOOST_FRAMEWORK_CPU_GPU_VERSIONS = ("1.2-1", "1.2-2")
4949

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

0 commit comments

Comments
 (0)