disable one test

Author: xadupre

_unittests/ut_torch_models/test_tiny_llms.py

@@ -14,7 +14,7 @@ def test_get_tiny_llm(self):
         model(**inputs)
 
     @ignore_warnings(UserWarning)
-    @requires_transformers("4.52")
+    @requires_transformers("4.53")
     def test_export_tiny_llm_1(self):
         data = get_tiny_llm()
         model, inputs = data["model"], data["inputs"]

onnx_diagnostic/torch_export_patches/eval/__init__.py

@@ -70,7 +70,7 @@ def evaluation(
 
     if isinstance(cases, (list, tuple)):
         all_cases = discover()
-        new_cases = []
+        new_cases = []  # type: ignore[var-annotated]
         for c in cases:
             if "*" in c or "?" in c:
                 # regex
@@ -552,8 +552,8 @@ def run_exporter(
         dict(
             success=1 if disc["abs"] < 0.1 else 0,
             model_cls=model.__class__,
-            exported=mod,
-            onnx=onx,
+            exported=mod,  # type: ignore[dict-item]
+            onnx=onx,  # type: ignore[dict-item]
         )
     )
     if disc["abs"] >= 0.1:

onnx_diagnostic/torch_export_patches/eval/model_cases.py

@@ -775,7 +775,7 @@ def forward(self, x):
 
 class CreateFromShapeThroughFunction(torch.nn.Module):
     @staticmethod
-    def add_one(dim: int) -> int:
+    def add_one(dim):
         return dim + 1
 
     def forward(self, x):