mypy

Author: xadupre

onnx_diagnostic/reference/torch_evaluator.py

@@ -206,5 +206,5 @@ def run(
             self.runtime_info[o].clean_value()
 
         if use_numpy:
-            return [None if a is None else a.detach().cpu().numpy() for a in res]
+            return [None if a is None else a.detach().cpu().numpy() for a in res]  # type: ignore[union-attr]
         return res  # type: ignore[return-value]