File tree Expand file tree Collapse file tree 2 files changed +15
-1
lines changed
unit/sagemaker/serve/builder Expand file tree Collapse file tree 2 files changed +15
-1
lines changed Original file line number Diff line number Diff 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"
Original file line number Diff line number Diff 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 ()
You can’t perform that action at this time.
0 commit comments