assign

xadupre · xadupre · commit 7a5b3be8de57 · 2025-06-12T09:40:38.000+02:00
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  # type: ignore[type-assignment]
+        shapes["interpolate_pos_encoding"] = None  # type: ignore[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 # type: ignore[type-assignment]`
	`72`	`+ shapes["interpolate_pos_encoding"] = None # type: ignore[assignment]`
`73`	`73`	`res = dict(inputs=inputs, dynamic_shapes=shapes)`
`74`	`74`	`if add_second_input:`
`75`	`75`	`res["inputs2"] = get_inputs(`