Skip to content

Commit 6e12169

Browse files
committed
Increase coverage
1 parent 90c9fbe commit 6e12169

File tree

2 files changed

+15
-1
lines changed

2 files changed

+15
-1
lines changed

tests/integ/sagemaker/serve/test_serve_transformers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,4 +127,4 @@ def test_pytorch_transformers_sagemaker_endpoint(
127127
logger.exception(caught_ex)
128128
assert (
129129
False
130-
), f"{caught_ex} was thrown when running pytorch transformers sagemaker endpoint test"
130+
), f"{caught_ex} thrown when running pytorch transformers sagemaker endpoint test"

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

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,3 +144,17 @@ def test_image_uri_override(
144144

145145
with self.assertRaises(ValueError) as _:
146146
model.deploy(mode=Mode.IN_PROCESS)
147+
148+
@patch(
149+
"sagemaker.huggingface.llm_utils.get_huggingface_model_metadata",
150+
return_value=None,
151+
)
152+
def test_failure_hf_md(self, mock_model_md):
153+
builder = ModelBuilder(
154+
model=mock_model_id,
155+
schema_builder=mock_schema_builder,
156+
mode=Mode.LOCAL_CONTAINER,
157+
)
158+
159+
with self.assertRaises(ValueError) as _:
160+
builder.build()

0 commit comments

Comments
 (0)