fix yield propagation

xadupre · xadupre · commit 55f3b3c1cf2f · 2025-05-16T21:04:34.000+02:00
diff --git a/onnx_diagnostic/torch_export_patches/onnx_export_errors.py b/onnx_diagnostic/torch_export_patches/onnx_export_errors.py
@@ -195,9 +195,9 @@ def torch_export_patches(
             verbose=verbose,
             patch=patch,
             custom_patches=custom_patches,
-        ):
+        ) as f:
             try:
-                yield
+                yield f
             finally:
                 pass
     elif not patch:
