File tree Expand file tree Collapse file tree
sagemaker-train/tests/unit/train/common_utils Expand file tree Collapse file tree Original file line number Diff line number Diff 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"""
You can’t perform that action at this time.
0 commit comments