fix mypy

Author: xadupre

onnx_diagnostic/torch_models/test_helper.py

@@ -181,17 +181,23 @@ def version_summary() -> Dict[str, Union[int, float, str]]:
 
 
 def _quiet_or_not_quiet(
-    quiet: bool, suffix: str, summary: Dict[str, Any], data: Dict[str, Any], fct: Callable
+    quiet: bool,
+    suffix: str,
+    summary: Dict[str, Any],
+    data: Optional[Dict[str, Any]],
+    fct: Callable,
 ) -> Any:
     begin = time.perf_counter()
     if quiet:
         try:
             return fct()
         except Exception as e:
             summary[f"ERR_{suffix}"] = str(e)
-            data[f"ERR_{suffix}"] = e
             summary[f"time_{suffix}"] = time.perf_counter() - begin
-            return summary, {}
+            if data is None:
+                return {f"ERR_{suffix}": e}
+            data[f"ERR_{suffix}"] = e
+            return None
     res = fct()
     summary[f"time_{suffix}"] = time.perf_counter() - begin
     return res