Skip to content

Commit 27cb74f

Browse files
authored
fix: add us-west-2 to Nova supported regions (#5782)
* fix: add us-west-2 to Nova supported regions in finetune_utils * test: update nova region test to reflect us-west-2 support
1 parent f99ae71 commit 27cb74f

2 files changed

Lines changed: 4 additions & 3 deletions

File tree

sagemaker-train/src/sagemaker/train/common_utils/finetune_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121

2222
# Region mappings for model availability
2323
OPEN_WEIGHTS_REGIONS = ['us-east-1', 'us-west-2', 'ap-northeast-1', 'eu-west-1'] # IAD, PDX, NRT, DUB
24-
NOVA_REGIONS = ['us-east-1'] # IAD only
24+
NOVA_REGIONS = ['us-east-1', 'us-west-2'] # IAD, PDX
2525
# Constants
2626
DEFAULT_REGION = "us-west-2"
2727

sagemaker-train/tests/unit/train/common_utils/test_finetune_utils.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -488,11 +488,12 @@ def test__validate_model_region_availability_nova_valid_region(self):
488488
"""Test Nova model validation passes for valid region"""
489489
# Should not raise any exception
490490
_validate_model_region_availability("nova-textgeneration-lite-v2", "us-east-1")
491+
_validate_model_region_availability("nova-textgeneration-lite-v2", "us-west-2")
491492

492493
def test__validate_model_region_availability_nova_invalid_region(self):
493494
"""Test Nova model validation fails for invalid region"""
494-
with pytest.raises(ValueError, match="Region 'us-west-2' does not support model customization"):
495-
_validate_model_region_availability("nova-textgeneration-lite-v2", "us-west-2")
495+
with pytest.raises(ValueError, match="Region 'eu-west-1' does not support model customization"):
496+
_validate_model_region_availability("nova-textgeneration-lite-v2", "eu-west-1")
496497

497498
def test__validate_model_region_availability_open_weights_valid_region(self):
498499
"""Test open weights model validation passes for valid region"""

0 commit comments

Comments
 (0)