Fix formatting

aws · May 15, 2024 · 749dc4a · 749dc4a
1 parent 3686dd9
commit 749dc4a
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/tests/unit/sagemaker/serve/builder/test_transformers_builder.py b/tests/unit/sagemaker/serve/builder/test_transformers_builder.py
@@ -115,7 +115,6 @@ def test_image_uri(
             model=mock_model_id,
             schema_builder=mock_schema_builder,
             mode=Mode.LOCAL_CONTAINER,
-            vpc_config=MOCK_VPC_CONFIG,
             image_config=MOCK_IMAGE_CONFIG,
         )
 
@@ -129,7 +128,6 @@ def test_image_uri(
         predictor = model.deploy(model_data_download_timeout=1800)
 
         assert model.image_config == MOCK_IMAGE_CONFIG
-        assert model.vpc_config == MOCK_VPC_CONFIG
         assert builder.env_vars["MODEL_LOADING_TIMEOUT"] == "1800"
         assert isinstance(predictor, TransformersLocalModePredictor)