Update image name tests to address new region

Author: Roja Reddy Sareddy

tests/unit/sagemaker/image_uris/expected_uris.py

@@ -116,3 +116,10 @@ def sagemaker_distribution_uri(repo, account, tag, processor, region=REGION):
     else:
         tag = f"{tag}-gpu"
     return IMAGE_URI_FORMAT.format(account, region, domain, repo, tag)
+
+def get_special_region_domain(region):
+    SPECIAL_REGIONS = {
+        "eu-isoe-west-1": ".cloud.adc-e.uk",
+        "eusc-de-east-1": ".amazonaws.eu"
+    }
+    return SPECIAL_REGIONS.get(region, ".amazonaws.com")

tests/unit/sagemaker/image_uris/test_dlc_frameworks.py

@@ -107,6 +107,11 @@ def test_dlc_framework_uris(load_config_and_file_name, scope):
                         region=region,
                         account=ACCOUNTS[region],
                     )
+                    # Handle special regions
+                    domain = expected_uris.get_special_region_domain(region)
+                    if domain != ".amazonaws.com":
+                        expected = expected.replace(".amazonaws.com", domain)
+
                     assert uri == expected
 
 

tests/unit/sagemaker/image_uris/test_graviton.py

@@ -43,6 +43,10 @@ def _test_graviton_framework_uris(
             region=region,
             container_version=container_version,
         )
+        # Handle special regions
+        domain = expected_uris.get_special_region_domain(region)
+        if domain != ".amazonaws.com":
+            expected = expected.replace(".amazonaws.com", domain)
         assert expected == uri
 
 

tests/unit/sagemaker/image_uris/test_huggingface_llm.py

@@ -111,6 +111,10 @@ def test_huggingface_uris(load_config):
                 HF_VERSIONS_MAPPING[device][version],
                 region=region,
             )
+            # Handle special regions
+            domain = expected_uris.get_special_region_domain(region)
+            if domain != ".amazonaws.com":
+                expected = expected.replace(".amazonaws.com", domain)
             assert expected == uri