mypy

Author: xadupre

onnx_diagnostic/reference/report_results_comparison.py

@@ -71,6 +71,6 @@ def report(
                     diff = self.max_diff(t, t2)
                 else:
                     diff = self.max_diff(tensor, t2)
-                res.append((name, held_key, diff))
+                res.append((name, held_key, diff))  # type: ignore[arg-type]
                 self.report_cmp[name, held_key] = diff
         return res

onnx_diagnostic/reference/torch_evaluator.py

@@ -544,9 +544,9 @@ def run(
                             zip(
                                 kernel.output,
                                 (
-                                    tuple((r.tensor if r else None) for r in res)
+                                    tuple((r.tensor if r else None) for r in res)  # type: ignore[attr-defined]
                                     if isinstance(res, tuple)
-                                    else ((res.tensor if res else None),)
+                                    else ((res.tensor if res else None),)  # type: ignore[attr-defined]
                                 ),
                             )
                         )