diff --git a/src/sagemaker/jumpstart/region_config.json b/src/sagemaker/jumpstart/region_config.json index 96209f9957..1820f71e69 100644 --- a/src/sagemaker/jumpstart/region_config.json +++ b/src/sagemaker/jumpstart/region_config.json @@ -119,6 +119,10 @@ "gated_content_bucket": "jumpstart-private-cache-prod-eu-west-3", "neo_content_bucket": "sagemaker-sd-models-prod-eu-west-3" }, + "eusc-de-east-1": { + "content_bucket": "jumpstart-cache-prod-eusc-de-east-1", + "gated_content_bucket": "jumpstart-private-cache-prod-eusc-de-east-1" + }, "il-central-1": { "content_bucket": "jumpstart-cache-prod-il-central-1", "gated_content_bucket": "jumpstart-private-cache-prod-il-central-1" diff --git a/src/sagemaker/jumpstart/utils.py b/src/sagemaker/jumpstart/utils.py index a9f18d381e..06566daed6 100644 --- a/src/sagemaker/jumpstart/utils.py +++ b/src/sagemaker/jumpstart/utils.py @@ -165,7 +165,7 @@ def get_jumpstart_content_bucket( except KeyError: formatted_launched_regions_str = get_jumpstart_launched_regions_message() raise ValueError( - f"Unable to get content bucket for Neo in {region} region. " + f"Unable to get content bucket for JumpStart in {region} region. " f"{formatted_launched_regions_str}" ) diff --git a/tests/unit/sagemaker/jumpstart/test_utils.py b/tests/unit/sagemaker/jumpstart/test_utils.py index 398327491a..2d0dc8c831 100644 --- a/tests/unit/sagemaker/jumpstart/test_utils.py +++ b/tests/unit/sagemaker/jumpstart/test_utils.py @@ -74,9 +74,18 @@ class TestBucketUtils(TestCase): def test_get_jumpstart_content_bucket(self): bad_region = "bad_region" assert bad_region not in JUMPSTART_REGION_NAME_SET - with pytest.raises(ValueError): + with pytest.raises( + ValueError, + match=r"^Unable to get content bucket for JumpStart in bad_region region\.", + ): utils.get_jumpstart_content_bucket(bad_region) + def test_get_jumpstart_content_bucket_eusc(self): + assert ( + utils.get_jumpstart_content_bucket("eusc-de-east-1") + == "jumpstart-cache-prod-eusc-de-east-1" + ) + def test_get_jumpstart_content_bucket_no_args(self): assert ( utils.get_jumpstart_content_bucket(JUMPSTART_DEFAULT_REGION_NAME) @@ -96,6 +105,12 @@ def test_get_jumpstart_gated_content_bucket(self): with pytest.raises(ValueError): utils.get_jumpstart_gated_content_bucket(bad_region) + def test_get_jumpstart_gated_content_bucket_eusc(self): + assert ( + utils.get_jumpstart_gated_content_bucket("eusc-de-east-1") + == "jumpstart-private-cache-prod-eusc-de-east-1" + ) + def test_get_jumpstart_gated_content_bucket_no_args(self): assert ( utils.get_jumpstart_gated_content_bucket(JUMPSTART_DEFAULT_REGION_NAME)