mypy

xadupre · xadupre · commit ea528c1b597a · 2025-10-03T12:38:33.000+02:00
diff --git a/onnx_diagnostic/torch_export_patches/eval/__init__.py b/onnx_diagnostic/torch_export_patches/eval/__init__.py
@@ -488,7 +488,7 @@ def _compares_on_one_example(
     disc.update(
         dict(
             success=1 if disc["abs"] < 0.1 else 0,
-            model_cls=model.__class__,
+            model_cls=model.__class__,  # type: ignore[dict-item]
             exported=mod,  # type: ignore[dict-item]
         )
     )

Original file line number	Diff line number	Diff line change
`@@ -488,7 +488,7 @@ def _compares_on_one_example(`
`488`	`488`	`disc.update(`
`489`	`489`	`dict(`
`490`	`490`	`success=1 if disc["abs"] < 0.1 else 0,`
`491`		`- model_cls=model.__class__,`
	`491`	`+ model_cls=model.__class__, # type: ignore[dict-item]`
`492`	`492`	`exported=mod, # type: ignore[dict-item]`
`493`	`493`	`)`
`494`	`494`	`)`