fix: fix mypy colpali (#533)

joein · web-flow · commit acec31277b78 · 2025-06-16T11:55:29.000+03:00
diff --git a/fastembed/late_interaction_multimodal/colpali.py b/fastembed/late_interaction_multimodal/colpali.py
@@ -175,7 +175,7 @@ def _preprocess_onnx_text_input(
     ) -> dict[str, NumpyArray]:
         onnx_input["input_ids"] = np.array(
             [
-                self.QUERY_MARKER_TOKEN_ID + input_ids[2:].tolist()
+                self.QUERY_MARKER_TOKEN_ID + input_ids[2:].tolist()  # type: ignore[index]
                 for input_ids in onnx_input["input_ids"]
             ]
         )

Original file line number	Diff line number	Diff line change
`@@ -175,7 +175,7 @@ def _preprocess_onnx_text_input(`
`175`	`175`	`) -> dict[str, NumpyArray]:`
`176`	`176`	`onnx_input["input_ids"] = np.array(`
`177`	`177`	`[`
`178`		`- self.QUERY_MARKER_TOKEN_ID + input_ids[2:].tolist()`
	`178`	`+ self.QUERY_MARKER_TOKEN_ID + input_ids[2:].tolist() # type: ignore[index]`
`179`	`179`	`for input_ids in onnx_input["input_ids"]`
`180`	`180`	`]`
`181`	`181`	`)`