code coverage

Author: xadupre

_unittests/ut_torch_onnx/test_sbs.py

@@ -46,7 +46,6 @@ def forward(self, x):
 
     @hide_stdout()
     @ignore_warnings((DeprecationWarning, FutureWarning, UserWarning))
-    @unittest.skipIf(to_onnx is None, "to_onnx not installed")
     def test_ep_onnx_sync_a(self):
         import torch
 
@@ -63,9 +62,10 @@ def forward(self, x):
         ep = torch.export.export(
             Model(), (x,), dynamic_shapes=({0: torch.export.Dim("batch")},)
         )
-        onx = torch.onnx.export(
-            Model(), (x,), dynamic_shapes=({0: torch.export.Dim("batch")},), dynamo=True
-        ).model_proto
+        epo = torch.onnx.export(
+            ep, (x,), dynamic_shapes=({0: torch.export.Dim("batch")},), dynamo=True
+        )
+        onx = epo.model_proto
         results = list(
             run_aligned(
                 ep,