fix mypy

titaiwangms · titaiwangms · commit df3ad9ba011c · 2025-09-23T21:25:51.000Z
diff --git a/onnx_diagnostic/tasks/text_generation.py b/onnx_diagnostic/tasks/text_generation.py
@@ -247,7 +247,7 @@ def get_inputs(
                 )
                 .to(torch.int64)
                 .expand((batch_size, -1)),
-                past_key_values=make_cache(
+                past_key_values=make_cache(  # type: ignore[operator]
                     [
                         (
                             torch.randn(

Original file line number	Diff line number	Diff line change
`@@ -247,7 +247,7 @@ def get_inputs(`
`247`	`247`	`)`
`248`	`248`	`.to(torch.int64)`
`249`	`249`	`.expand((batch_size, -1)),`
`250`		`- past_key_values=make_cache(`
	`250`	`+ past_key_values=make_cache( # type: ignore[operator]`
`251`	`251`	`[`
`252`	`252`	`(`
`253`	`253`	`torch.randn(`