pymp

Author: xadupre

onnx_diagnostic/torch_models/test_helper.py

@@ -1126,7 +1126,7 @@ def call_torch_export_model_builder(
     ), f"unexpected value for optimization={optimization}, none is available"
     precision = data.get("model_dtype", "fp32")
     provider = data.get("model_device", "cpu")
-    summary = {}
+    summary: Dict[str, Any] = {}
 
     epo = _quiet_or_not_quiet(
         quiet,

pyproject.toml

@@ -32,6 +32,10 @@ disable_error_code = ["arg-type", "call-overload", "index"]
 module = ["onnx_diagnostic.helpers.helper"]
 disable_error_code = ["arg-type", "assignment", "attr-defined", "call-overload", "misc", "name-defined", "union-attr"]
 
+[[tool.mypy.overrides]]
+module = ["onnx_diagnostic.helpers.model_builder_helper"]
+disable_error_code = ["attr-defined", "import-untyped", "name-defined", "var-annotated"]
+
 [[tool.mypy.overrides]]
 module = ["onnx_diagnostic.helpers.onnx_helper"]
 disable_error_code = ["assignment", "arg-type", "name-defined", "union-attr"]