test'

Author: xadupre

_doc/conf.py

@@ -116,6 +116,7 @@ def linkcode_resolve(domain, info):
     ("py:class", "Argument"),
     ("py:class", "default=sklearn.utils.metadata_routing.UNCHANGED"),
     ("py:class", "ModelProto"),
+    ("py:class", "Model"),
     ("py:class", "Module"),
     ("py:class", "np.ndarray"),
     ("py:class", "onnxscript.ir.Tuple"),

_doc/status/patches_coverage.rst

@@ -28,7 +28,7 @@ for transformers.
     import onnx_diagnostic.torch_export_patches.patches.patch_transformers as p
 
     for name, cls in p.__dict__.items():
-        if name.startswith("patched_"):
+        if name.startswith("patched_") and hasattr(cls, "_PATCHES_"):
             print(f"{cls._PATCHED_CLASS_.__name__}: {', '.join(cls._PATCHES_)}")
 
 Half Automated Rewrites for Control Flows

_unittests/ut_torch_models/test_test_helpers.py

@@ -9,6 +9,7 @@
     requires_torch,
     requires_experimental,
     requires_onnxscript,
+    requires_transformers,
 )
 from onnx_diagnostic.torch_models.test_helper import (
     get_inputs_for_task,
@@ -187,6 +188,7 @@ def test_filter_inputs(self):
     @requires_torch("2.7")
     @hide_stdout()
     @ignore_warnings(FutureWarning)
+    @requires_transformers("4.51")
     def test_validate_model_modelbuilder(self):
         mid = "arnir0/Tiny-LLM"
         summary, data = validate_model(