Update test_runtime_ort.py

zhiqwang · Aug 3, 2023 · 546b852 · 546b852
1 parent a2790eb
commit 546b852
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_runtime_ort.py b/test/test_runtime_ort.py
@@ -132,5 +132,5 @@ def test_onnx_export_misbatch(self, arch):
             self.run_model(model, [[img_one, img_two], [img_two, img_one, img_dummy]])
 
         # Test exported model on images of misbatch
-        with pytest.raises(ValueError, match="Model requires 3 inputs. Input Feed contains 2"):
+        with pytest.raises(ValueError, match="Required inputs (['image3']) are missing from input feed"):
             self.run_model(model, [[img_two, img_one, img_dummy], [img_one, img_two]])