Skip to content

Commit e8ffe70

Browse files
author
Chuyang Deng
committed
fix: blazingtext tag and upgrade sparkml version
1 parent d540006 commit e8ffe70

File tree

3 files changed

+32
-2
lines changed

3 files changed

+32
-2
lines changed

src/sagemaker/image_uri_config/blazingtext.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
{
22
"scope": ["inference", "training"],
33
"versions": {
4-
"1": {
4+
"latest": {
55
"registries": {
66
"af-south-1": "455444449433",
77
"ap-east-1": "286214385809",

src/sagemaker/image_uri_config/sparkml-serving.json

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,35 @@
2929
"us-west-2": "246618743249"
3030
},
3131
"repository": "sagemaker-sparkml-serving"
32+
},
33+
"2.4": {
34+
"registries": {
35+
"af-south-1": "510948584623",
36+
"ap-east-1": "651117190479",
37+
"ap-northeast-1": "354813040037",
38+
"ap-northeast-2": "366743142698",
39+
"ap-south-1": "720646828776",
40+
"ap-southeast-1": "121021644041",
41+
"ap-southeast-2": "783357654285",
42+
"ca-central-1": "341280168497",
43+
"cn-north-1": "450853457545",
44+
"cn-northwest-1": "451049120500",
45+
"eu-central-1": "492215442770",
46+
"eu-north-1": "662702820516",
47+
"eu-west-1": "141502667606",
48+
"eu-west-2": "764974769150",
49+
"eu-west-3": "659782779980",
50+
"eu-south-1": "978288397137",
51+
"me-south-1": "801668240914",
52+
"sa-east-1": "737474898029",
53+
"us-east-1": "683313688378",
54+
"us-east-2": "257758044811",
55+
"us-gov-west-1": "414596584902",
56+
"us-iso-east-1": "833128469047",
57+
"us-west-1": "746614075791",
58+
"us-west-2": "246618743249"
59+
},
60+
"repository": "sagemaker-sparkml-serving"
3261
}
3362
}
3463
}

tests/unit/sagemaker/image_uris/test_algos.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,8 @@ def test_algo_uris(algo):
177177

178178
for region in regions.regions():
179179
uri = image_uris.retrieve(algo, region)
180-
assert expected_uris.algo_uri(algo, accounts[region], region) == uri
180+
version = "latest" if algo == "blazingtext" else 1
181+
assert expected_uris.algo_uri(algo, accounts[region], region, version) == uri
181182

182183

183184
def test_lda():

0 commit comments

Comments
 (0)