mypy

xadupre · xadupre · commit 2167823ee15a · 2025-06-12T09:36:39.000+02:00
diff --git a/CHANGELOGS.rst b/CHANGELOGS.rst
@@ -4,10 +4,10 @@ Change Logs
 0.7.0
 +++++
 
-* :pr:`143`: compares intermediate results,
-  support for second inputs with different dimension,
+* :pr:`144`: support for second inputs with different dimension,
   rename test_helper into validate,
   support ``interpolate_pos_encoding`` for ``VitModel``
+* :pr:`143`: compares intermediate results,
 
 0.6.3
 +++++
diff --git a/onnx_diagnostic/tasks/image_classification.py b/onnx_diagnostic/tasks/image_classification.py
@@ -69,7 +69,7 @@ def get_inputs(
     )
     if model.__class__.__name__ == "ViTForImageClassification":
         inputs["interpolate_pos_encoding"] = True
-        shapes["interpolate_pos_encoding"] = None
+        shapes["interpolate_pos_encoding"] = None  # type: ignore[type-assignment]
     res = dict(inputs=inputs, dynamic_shapes=shapes)
     if add_second_input:
         res["inputs2"] = get_inputs(

Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,7 @@ def get_inputs(`
`69`	`69`	`)`
`70`	`70`	`if model.__class__.__name__ == "ViTForImageClassification":`
`71`	`71`	`inputs["interpolate_pos_encoding"] = True`
`72`		`- shapes["interpolate_pos_encoding"] = None`
	`72`	`+ shapes["interpolate_pos_encoding"] = None # type: ignore[type-assignment]`
`73`	`73`	`res = dict(inputs=inputs, dynamic_shapes=shapes)`
`74`	`74`	`if add_second_input:`
`75`	`75`	`res["inputs2"] = get_inputs(`