Enable outputs validation for default export.

mfuntowicz · mfuntowicz · commit 19a3a3351597 · 2021-06-29T17:22:55.000+02:00
diff --git a/tests/test_onnx_v2.py b/tests/test_onnx_v2.py
@@ -211,7 +211,13 @@ def test_pytorch_export_default(self):
                 onnx_config = onnx_config_class.default(model.config)
 
                 with NamedTemporaryFile("w") as output:
-                    convert_pytorch(tokenizer, model, onnx_config, DEFAULT_ONNX_OPSET, Path(output.name))
+                    onnx_inputs, onnx_outputs = \
+                        convert_pytorch(tokenizer, model, onnx_config, DEFAULT_ONNX_OPSET, Path(output.name))
+
+                    try:
+                        validate_model_outputs(onnx_config, tokenizer, model, Path(output.name), onnx_outputs, 1e-5)
+                    except ValueError as ve:
+                        self.fail(f"{name} -> {ve}")
 
     @slow
     @require_torch
