|
13 | 13 | from __future__ import absolute_import |
14 | 14 |
|
15 | 15 | 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 |
17 | 17 |
|
18 | 18 | ACCOUNTS = { |
| 19 | + "af-south-1": "875698925577", |
19 | 20 | "ap-east-1": "001633400207", |
20 | 21 | "ap-northeast-1": "574779866223", |
21 | 22 | "ap-northeast-2": "709848358524", |
|
27 | 28 | "cn-northwest-1": "453252182341", |
28 | 29 | "eu-central-1": "048819808253", |
29 | 30 | "eu-north-1": "895015795356", |
| 31 | + "eu-south-1": "933208885752", |
30 | 32 | "eu-west-1": "468650794304", |
31 | 33 | "eu-west-2": "749857270468", |
32 | 34 | "eu-west-3": "680080141114", |
|
40 | 42 |
|
41 | 43 |
|
42 | 44 | 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) |
46 | 47 |
|
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