Skip to content

Commit 2400c80

Browse files
committed
test: update nova region test to reflect us-west-2 support
1 parent b1c512f commit 2400c80

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

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)