Skip to content

Commit aee2dba

Browse files
committed
fix(jumpstart): add eusc-de-east-1 buckets and name JumpStart in the bucket error
`JumpStartModel` for `deepseek-llm-r1-distill-llama-8b` in `eusc-de-east-1` failed with `ValueError: Unable to get content bucket for Neo in eusc-de-east-1 region.` The region configuration had no `eusc-de-east-1` entry, and the JumpStart bucket accessor named Neo in its error. The model does not use Neo. Add the public and gated JumpStart buckets for `eusc-de-east-1`. Name JumpStart in the error. --- X-AI-Prompt: What is the minimal source change that fixes the EU Sovereign Cloud JumpStart error? X-AI-Tool: claude-code
1 parent 58e4ca8 commit aee2dba

3 files changed

Lines changed: 21 additions & 2 deletions

File tree

‎src/sagemaker/jumpstart/region_config.json‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,10 @@
119119
"gated_content_bucket": "jumpstart-private-cache-prod-eu-west-3",
120120
"neo_content_bucket": "sagemaker-sd-models-prod-eu-west-3"
121121
},
122+
"eusc-de-east-1": {
123+
"content_bucket": "jumpstart-cache-prod-eusc-de-east-1",
124+
"gated_content_bucket": "jumpstart-private-cache-prod-eusc-de-east-1"
125+
},
122126
"il-central-1": {
123127
"content_bucket": "jumpstart-cache-prod-il-central-1",
124128
"gated_content_bucket": "jumpstart-private-cache-prod-il-central-1"

‎src/sagemaker/jumpstart/utils.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ def get_jumpstart_content_bucket(
165165
except KeyError:
166166
formatted_launched_regions_str = get_jumpstart_launched_regions_message()
167167
raise ValueError(
168-
f"Unable to get content bucket for Neo in {region} region. "
168+
f"Unable to get content bucket for JumpStart in {region} region. "
169169
f"{formatted_launched_regions_str}"
170170
)
171171

‎tests/unit/sagemaker/jumpstart/test_utils.py‎

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,9 +74,18 @@ class TestBucketUtils(TestCase):
7474
def test_get_jumpstart_content_bucket(self):
7575
bad_region = "bad_region"
7676
assert bad_region not in JUMPSTART_REGION_NAME_SET
77-
with pytest.raises(ValueError):
77+
with pytest.raises(
78+
ValueError,
79+
match=r"^Unable to get content bucket for JumpStart in bad_region region\.",
80+
):
7881
utils.get_jumpstart_content_bucket(bad_region)
7982

83+
def test_get_jumpstart_content_bucket_eusc(self):
84+
assert (
85+
utils.get_jumpstart_content_bucket("eusc-de-east-1")
86+
== "jumpstart-cache-prod-eusc-de-east-1"
87+
)
88+
8089
def test_get_jumpstart_content_bucket_no_args(self):
8190
assert (
8291
utils.get_jumpstart_content_bucket(JUMPSTART_DEFAULT_REGION_NAME)
@@ -96,6 +105,12 @@ def test_get_jumpstart_gated_content_bucket(self):
96105
with pytest.raises(ValueError):
97106
utils.get_jumpstart_gated_content_bucket(bad_region)
98107

108+
def test_get_jumpstart_gated_content_bucket_eusc(self):
109+
assert (
110+
utils.get_jumpstart_gated_content_bucket("eusc-de-east-1")
111+
== "jumpstart-private-cache-prod-eusc-de-east-1"
112+
)
113+
99114
def test_get_jumpstart_gated_content_bucket_no_args(self):
100115
assert (
101116
utils.get_jumpstart_gated_content_bucket(JUMPSTART_DEFAULT_REGION_NAME)

0 commit comments

Comments
 (0)