Skip to content

Commit 749dc4a

Browse files
committed
Fix formatting
1 parent 3686dd9 commit 749dc4a

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

tests/unit/sagemaker/serve/builder/test_transformers_builder.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,6 @@ def test_image_uri(
115115
model=mock_model_id,
116116
schema_builder=mock_schema_builder,
117117
mode=Mode.LOCAL_CONTAINER,
118-
vpc_config=MOCK_VPC_CONFIG,
119118
image_config=MOCK_IMAGE_CONFIG,
120119
)
121120

@@ -129,7 +128,6 @@ def test_image_uri(
129128
predictor = model.deploy(model_data_download_timeout=1800)
130129

131130
assert model.image_config == MOCK_IMAGE_CONFIG
132-
assert model.vpc_config == MOCK_VPC_CONFIG
133131
assert builder.env_vars["MODEL_LOADING_TIMEOUT"] == "1800"
134132
assert isinstance(predictor, TransformersLocalModePredictor)
135133

0 commit comments

Comments
 (0)