diff --git a/sagemaker-core/src/sagemaker/core/jumpstart/utils.py b/sagemaker-core/src/sagemaker/core/jumpstart/utils.py index 92109f203e..f4413947fa 100644 --- a/sagemaker-core/src/sagemaker/core/jumpstart/utils.py +++ b/sagemaker-core/src/sagemaker/core/jumpstart/utils.py @@ -204,7 +204,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/sagemaker-core/tests/unit/test_jumpstart_utils.py b/sagemaker-core/tests/unit/test_jumpstart_utils.py index 08eb20ccdc..4b15b11f4e 100644 --- a/sagemaker-core/tests/unit/test_jumpstart_utils.py +++ b/sagemaker-core/tests/unit/test_jumpstart_utils.py @@ -491,7 +491,7 @@ def test_get_jumpstart_content_bucket_with_override(self): def test_get_jumpstart_content_bucket_invalid_region(self): """Test with invalid region""" with patch.object(constants, "JUMPSTART_REGION_NAME_TO_LAUNCHED_REGION_DICT", {}): - with pytest.raises(ValueError, match="Unable to get content bucket for Neo"): + with pytest.raises(ValueError, match="Unable to get content bucket for JumpStart"): utils.get_jumpstart_content_bucket("invalid-region")