improves CI

Author: xadupre

.github/workflows/ci.yml

@@ -16,7 +16,7 @@ jobs:
       matrix:
         os: [ubuntu-latest]
         python: ['3.11', '3.12']
-        transformers: ['4.48', '4.50', 'main']
+        transformers: ['4.48.3', '4.50.3', 'main']
         torch: ['2.6', 'main']
 
     steps:

_doc/examples/plot_export_locate_issue.py

@@ -85,7 +85,7 @@ def forward(self, x: torch.Tensor, ys: list[torch.Tensor]):
 # to replace torch implementation by a new one raising the exception
 # mentioned in previous section.
 
-with bypass_export_some_errors(stop_if_static=True, verbose=1):
+with bypass_export_some_errors(stop_if_static=1, verbose=1):
     try:
         torch.export.export(model, inputs, dynamic_shapes=dyn_shapes)
     except (AssertionError, torch._dynamo.exc.TorchRuntimeError) as e:

_unittests/ut_torch_models/test_test_helpers.py

@@ -1,5 +1,7 @@
 import copy
 import unittest
+import packaging.version as pv
+import transformers
 from onnx_diagnostic.ext_test_case import ExtTestCase, hide_stdout, ignore_warnings
 from onnx_diagnostic.torch_models.test_helper import (
     get_inputs_for_task,
@@ -62,6 +64,7 @@ def test_validate_model_onnx(self):
             exporter="onnx-dynamo",
             dump_folder="dump_test_validate_model_onnx",
             patch=True,
+            stop_if_static=pv.Version(transformers.__version__) >= pv.Version("4.50"),
         )
         self.assertIsInstance(summary, dict)
         self.assertIsInstance(data, dict)

onnx_diagnostic/_command_lines_parser.py

@@ -267,6 +267,7 @@ def get_parser_validate() -> ArgumentParser:
     parser.add_argument(
         "--stop-if-static",
         default=0,
+        type=int,
         help="raises an exception if a dynamic dimension becomes static",
     )
     parser.add_argument(

onnx_diagnostic/torch_models/test_helper.py

@@ -202,7 +202,7 @@ def validate_model(
     optimization: Optional[str] = None,
     quiet: bool = False,
     patch: bool = False,
-    stop_if_static: bool = True,
+    stop_if_static: int = 1,
     dump_folder: Optional[str] = None,
     drop_inputs: Optional[List[str]] = None,
 ) -> Tuple[Dict[str, Union[int, float, str]], Dict[str, Any]]: