fix

Author: xadupre

_unittests/ut_torch_export_patches/test_dynamic_class.py

@@ -8,6 +8,7 @@
     ignore_warnings,
     hide_stdout,
     requires_torch,
+    has_transformers,
 )
 from onnx_diagnostic.helpers import string_type
 from onnx_diagnostic.cache_helpers import make_dynamic_cache
@@ -21,7 +22,8 @@ class TestOnnxExportErrors(ExtTestCase):
     @ignore_warnings(UserWarning)
     @hide_stdout()
     def test_export_dynamic_cache_update(self):
-        for strict in self.subloop([True, False], verbose=1):
+        values = [True, False] if has_transformers("4.50") else [False]
+        for strict in self.subloop(values, verbose=1):
 
             class SubModelCache(torch.nn.Module):
                 def forward(self, cache):

_unittests/ut_torch_models/test_tiny_llms_onnx.py

@@ -20,7 +20,7 @@
 
 class TestTinyLlmOnnx(ExtTestCase):
     @ignore_warnings((UserWarning, DeprecationWarning, FutureWarning))
-    @requires_transformers("4.50.9999")
+    @requires_transformers("4.52.9999")
     @hide_stdout()
     def test_onnx_export_tiny_llm_official(self):
         data = get_tiny_llm()
@@ -53,9 +53,10 @@ def test_onnx_export_tiny_llm_xdbg(self):
         self.assertEqual(
             {"attention_mask", "past_key_values", "input_ids", "position_ids"}, set(inputs)
         )
-        onx = to_onnx(
-            model, (), kwargs=inputs, dynamic_shapes=data["dynamic_shapes"], verbose=1
-        )
+        with bypass_export_some_errors(patch_transformers=True):
+            onx = to_onnx(
+                model, (), kwargs=inputs, dynamic_shapes=data["dynamic_shapes"], verbose=1
+            )
         self.assert_onnx_disc(
             inspect.currentframe().f_code.co_name, onx, model, inputs, verbose=1
         )

onnx_diagnostic/ext_test_case.py

@@ -1097,7 +1097,7 @@ def _debug(self):
         return os.environ.get("DEBUG") in BOOLEAN_VALUES
 
     def subloop(self, *args, verbose: int = 0):
-        "Loops over elements and calls :meth:`unittests.TextCase.subTest`."
+        "Loops over elements and calls :meth:`unittests.TestCase.subTest`."
         if len(args) == 1:
             for it in args[0]:
                 with self.subTest(case=it):