We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3686dd9 commit 749dc4aCopy full SHA for 749dc4a
tests/unit/sagemaker/serve/builder/test_transformers_builder.py
@@ -115,7 +115,6 @@ def test_image_uri(
115
model=mock_model_id,
116
schema_builder=mock_schema_builder,
117
mode=Mode.LOCAL_CONTAINER,
118
- vpc_config=MOCK_VPC_CONFIG,
119
image_config=MOCK_IMAGE_CONFIG,
120
)
121
@@ -129,7 +128,6 @@ def test_image_uri(
129
128
predictor = model.deploy(model_data_download_timeout=1800)
130
131
assert model.image_config == MOCK_IMAGE_CONFIG
132
- assert model.vpc_config == MOCK_VPC_CONFIG
133
assert builder.env_vars["MODEL_LOADING_TIMEOUT"] == "1800"
134
assert isinstance(predictor, TransformersLocalModePredictor)
135
0 commit comments