mypy

Author: xadupre

onnx_diagnostic/reference/report_results_comparison.py

@@ -55,12 +55,12 @@ def report(
         the results of the comparison. The function also collects the results
         into a dictionary the user can retrieve later.
         """
-        res = []
+        res: List[Tuple[str, ReportKeyNameType, Dict[str, Union[float, str]]]] = []
         for name, tensor in outputs.items():
             key = self.key(tensor)
             if key not in self.mapping:
                 continue
-            cache = {}
+            cache: Dict["torch.device", "torch.Tensor"] = {}  # noqa: F821, UP037
             for held_key in self.mapping[key]:
                 t2 = self.tensors[held_key]
                 if hasattr(t2, "device") and hasattr(tensor, "device"):

onnx_diagnostic/reference/torch_evaluator.py

@@ -544,9 +544,9 @@ def run(
                             zip(
                                 kernel.output,
                                 (
-                                    tuple(r.tensor for r in res)
+                                    tuple((r.tensor if r else None) for r in res)
                                     if isinstance(res, tuple)
-                                    else (res.tensor,)
+                                    else ((res.tensor if res else None),)
                                 ),
                             )
                         )

pyproject.toml

@@ -44,6 +44,10 @@ disable_error_code = ["assignment", "arg-type", "name-defined", "union-attr"]
 module = ["onnx_diagnostic.helpers.ort_session"]
 disable_error_code = ["union-attr"]
 
+[[tool.mypy.overrides]]
+module = ["onnx_diagnostic.reference.report_results_comparison"]
+disable_error_code = ["name-defined"]
+
 [[tool.mypy.overrides]]
 module = ["onnx_diagnostic.reference.torch_ops.*"]
 disable_error_code = ["override"]