Skip to content

Commit 5906311

Browse files
tejaschumbalkarknikure
authored andcommitted
ignore runs in specific regions
1 parent bfa3839 commit 5906311

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

tests/unit/sagemaker/image_uris/test_dlc_frameworks.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@
1717
from sagemaker import image_uris
1818
from tests.unit.sagemaker.image_uris import expected_uris
1919

20+
import pytest
21+
2022
INSTANCE_TYPES_AND_PROCESSORS = (("ml.c4.xlarge", "cpu"), ("ml.p2.xlarge", "gpu"))
2123
RENEWED_PYTORCH_INSTANCE_TYPES_AND_PROCESSORS = (("ml.c4.xlarge", "cpu"), ("ml.g4dn.xlarge", "gpu"))
2224
REGION = "us-west-2"
@@ -85,6 +87,14 @@ def _test_image_uris(
8587
assert expected == uri
8688

8789
for region in SAGEMAKER_ALTERNATE_REGION_ACCOUNTS.keys():
90+
if (
91+
scope == "training"
92+
and framework == "tensorflow"
93+
and Version(fw_version) == Version("2.12")
94+
):
95+
if region in ["cn-north-1", "cn-northwest-1", "us-iso-east-1", "us-isob-east-1"]:
96+
pytest.skip(f"TF 2.12 SM DLC is not available in {region} region")
97+
8898
uri = image_uris.retrieve(region=region, instance_type="ml.c4.xlarge", **base_args)
8999

90100
expected = expected_fn(region=region, **expected_fn_args)

0 commit comments

Comments
 (0)