Skip to content

Commit fc6ab19

Browse files
author
Roja Reddy Sareddy
committed
Update image name tests to address new region
1 parent c7cdf3b commit fc6ab19

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

tests/unit/sagemaker/image_uris/expected_uris.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -117,9 +117,7 @@ def sagemaker_distribution_uri(repo, account, tag, processor, region=REGION):
117117
tag = f"{tag}-gpu"
118118
return IMAGE_URI_FORMAT.format(account, region, domain, repo, tag)
119119

120+
120121
def get_special_region_domain(region):
121-
SPECIAL_REGIONS = {
122-
"eu-isoe-west-1": ".cloud.adc-e.uk",
123-
"eusc-de-east-1": ".amazonaws.eu"
124-
}
125-
return SPECIAL_REGIONS.get(region, ".amazonaws.com")
122+
SPECIAL_REGIONS = {"eu-isoe-west-1": ".cloud.adc-e.uk", "eusc-de-east-1": ".amazonaws.eu"}
123+
return SPECIAL_REGIONS.get(region, ".amazonaws.com")

0 commit comments

Comments
 (0)