api

Author: xadupre

_unittests/ut_export/test_api.py

@@ -6,6 +6,7 @@
     hide_stdout,
     has_transformers,
     ignore_warnings,
+    requires_transformers,
 )
 from onnx_diagnostic.helpers import max_diff
 from onnx_diagnostic.helpers.torch_helper import torch_deepcopy
@@ -43,6 +44,7 @@ def forward(self, x, y):
 
     @hide_stdout()
     @ignore_warnings(FutureWarning)
+    @requires_transformers("4.50")
     def test_tiny_llm_to_onnx(self):
         import onnxruntime