fix ut

Author: xadupre

_doc/api/torch_models/index.rst

@@ -0,0 +1,12 @@
+onnx_diagnostic.torch_models
+============================
+
+.. toctree::
+    :maxdepth: 1
+    :caption: submodules
+
+    llms
+
+.. automodule:: onnx_diagnostic.torch_models
+    :members:
+    :no-undoc-members:

_doc/api/torch_models/llms.rst

@@ -0,0 +1,7 @@
+
+onnx_diagnostic.torch_models.llms
+=================================
+
+.. automodule:: onnx_diagnostic.torch.models.llms
+    :members:
+    :no-undoc-members:

_unittests/ut_xrun_doc/test_documentation_examples.py

@@ -5,7 +5,8 @@
 import subprocess
 import time
 from onnx_diagnostic import __file__ as onnx_diagnostic_file
-from onnx_diagnostic.ext_test_case import ExtTestCase, is_windows
+from onnx_diagnostic.ext_test_case import ExtTestCase, is_windows, has_transformers
+
 
 VERBOSE = 0
 ROOT = os.path.realpath(os.path.abspath(os.path.join(onnx_diagnostic_file, "..", "..")))
@@ -69,6 +70,13 @@ def add_test_methods(cls):
                 continue
             reason = None
 
+            if (
+                not reason
+                and name in {"plot_export_tiny_llm.py"}
+                and not has_transformers("4.50")
+            ):
+                reason = "transformers<4.50"
+
             if reason:
 
                 @unittest.skip(reason)

onnx_diagnostic/ext_test_case.py

@@ -774,13 +774,21 @@ def requires_sklearn(version: str, msg: str = "") -> Callable:
 
 
 def has_torch(version: str) -> bool:
-    "Returns True if torch verions is higher."
+    "Returns True if torch transformers is higher."
     import packaging.version as pv
     import torch
 
     return pv.Version(torch.__version__) >= pv.Version(version)
 
 
+def has_transformers(version: str) -> bool:
+    "Returns True if transformers version is higher."
+    import packaging.version as pv
+    import transformers
+
+    return pv.Version(transformers.__version__) >= pv.Version(version)
+
+
 def requires_torch(version: str, msg: str = "") -> Callable:
     """Skips a unit test if :epkg:`pytorch` is not recent enough."""
     import packaging.version as pv