We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent abd6852 commit 31eed78Copy full SHA for 31eed78
tests/unit/sagemaker/serve/utils/test_hardware_detector.py
@@ -120,3 +120,7 @@ def test_total_inference_model_size_mib(
120
121
with pytest.raises(ValueError):
122
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