Skip to content

Commit 31eed78

Browse files
committed
Add coverage for import error
1 parent abd6852 commit 31eed78

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/unit/sagemaker/serve/utils/test_hardware_detector.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -120,3 +120,7 @@ def test_total_inference_model_size_mib(
120120

121121
with pytest.raises(ValueError):
122122
hardware_detector._total_inference_model_size_mib("stable-diffusion", "float32")
123+
124+
mock_parser.side_effect = ImportError
125+
with pytest.raises(ValueError):
126+
hardware_detector._total_inference_model_size_mib("stable-diffusion", "float32")

0 commit comments

Comments
 (0)