Skip to content

Commit ca86159

Browse files
feature: add model monitor image accounts for af-south-1 and eu-south-1 (#1883)
1 parent c0e280e commit ca86159

File tree

2 files changed

+9
-6
lines changed

2 files changed

+9
-6
lines changed

src/sagemaker/image_uri_config/model-monitor.json

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
"versions": {
44
"": {
55
"registries": {
6+
"af-south-1": "875698925577",
67
"ap-east-1": "001633400207",
78
"ap-northeast-1": "574779866223",
89
"ap-northeast-2": "709848358524",
@@ -14,6 +15,7 @@
1415
"cn-northwest-1": "453252182341",
1516
"eu-central-1": "048819808253",
1617
"eu-north-1": "895015795356",
18+
"eu-south-1": "933208885752",
1719
"eu-west-1": "468650794304",
1820
"eu-west-2": "749857270468",
1921
"eu-west-3": "680080141114",

tests/unit/sagemaker/image_uris/test_model_monitor.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,10 @@
1313
from __future__ import absolute_import
1414

1515
from sagemaker import image_uris
16-
from tests.unit.sagemaker.image_uris import expected_uris, regions
16+
from tests.unit.sagemaker.image_uris import expected_uris
1717

1818
ACCOUNTS = {
19+
"af-south-1": "875698925577",
1920
"ap-east-1": "001633400207",
2021
"ap-northeast-1": "574779866223",
2122
"ap-northeast-2": "709848358524",
@@ -27,6 +28,7 @@
2728
"cn-northwest-1": "453252182341",
2829
"eu-central-1": "048819808253",
2930
"eu-north-1": "895015795356",
31+
"eu-south-1": "933208885752",
3032
"eu-west-1": "468650794304",
3133
"eu-west-2": "749857270468",
3234
"eu-west-3": "680080141114",
@@ -40,9 +42,8 @@
4042

4143

4244
def test_model_monitor():
43-
for region in regions.regions():
44-
if region in ACCOUNTS.keys():
45-
uri = image_uris.retrieve("model-monitor", region=region)
45+
for region in ACCOUNTS.keys():
46+
uri = image_uris.retrieve("model-monitor", region=region)
4647

47-
expected = expected_uris.monitor_uri(ACCOUNTS[region], region)
48-
assert expected == uri
48+
expected = expected_uris.monitor_uri(ACCOUNTS[region], region)
49+
assert expected == uri

0 commit comments

Comments
 (0)