From f29920d67029b319816c68075337a798340573ab Mon Sep 17 00:00:00 2001 From: George Panchuk Date: Mon, 16 Jun 2025 12:54:07 +0400 Subject: [PATCH] fix: fix mypy colpali --- fastembed/late_interaction_multimodal/colpali.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastembed/late_interaction_multimodal/colpali.py b/fastembed/late_interaction_multimodal/colpali.py index c6b98e23e..9fab95359 100644 --- 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"] ] )