adapt test

echarlaix · echarlaix · commit 9fd663ea9b25 · 2025-02-07T19:30:11.000+01:00
diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
@@ -383,8 +383,7 @@ def test_load_stable_diffusion_model_unknown_provider(self):
             )
 
     def test_load_model_from_hub_without_onnx_model(self):
-        with self.assertRaises(FileNotFoundError):
-            ORTModel.from_pretrained(self.FAIL_ONNX_MODEL_ID)
+        ORTModel.from_pretrained(self.FAIL_ONNX_MODEL_ID)
 
     def test_model_on_cpu(self):
         model = ORTModel.from_pretrained(self.ONNX_MODEL_ID)

Original file line number	Diff line number	Diff line change
`@@ -383,8 +383,7 @@ def test_load_stable_diffusion_model_unknown_provider(self):`
`383`	`383`	`)`
`384`	`384`
`385`	`385`	`def test_load_model_from_hub_without_onnx_model(self):`
`386`		`- with self.assertRaises(FileNotFoundError):`
`387`		`- ORTModel.from_pretrained(self.FAIL_ONNX_MODEL_ID)`
	`386`	`+ ORTModel.from_pretrained(self.FAIL_ONNX_MODEL_ID)`
`388`	`387`
`389`	`388`	`def test_model_on_cpu(self):`
`390`	`389`	`model = ORTModel.from_pretrained(self.ONNX_MODEL_ID)`