Skip to content

Commit addb37b

Browse files
authored
fix: set sagemaker_config=None on mock session in test_from_jumpstart_config_applies_volume_size (#5882)
fix unit test sm-serve
1 parent 657864a commit addb37b

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

sagemaker-serve/tests/unit/test_model_builder_coverage_boost.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -424,6 +424,7 @@ def test_from_jumpstart_config_applies_volume_size(self, mock_deploy_kwargs):
424424

425425
mock_session = Mock()
426426
mock_session.boto_region_name = "us-west-2"
427+
mock_session.sagemaker_config = None
427428

428429
mb = ModelBuilder.from_jumpstart_config(
429430
jumpstart_config=js_config,

0 commit comments

Comments
 (0)