Fix mypy errors

Author: kacperlukawski

fastembed/late_interaction_multimodal/onnx_multimodal_model.py

@@ -184,15 +184,18 @@ def onnx_embed_image(self, images: list[ImageInput], **kwargs: Any) -> OnnxOutpu
             if isinstance(processed[0], list):
                 encoded, attention_mask, metadata = self._process_nested_patches(processed)
             else:
-                encoded, attention_mask, metadata = self._process_flat_images(processed, len(images))
+                encoded, attention_mask, metadata = self._process_flat_images(
+                    processed,  # type: ignore[arg-type]
+                    len(images),
+                )
 
         onnx_input = {"pixel_values": encoded, "attention_mask": attention_mask}
         onnx_input = self._preprocess_onnx_image_input(onnx_input, **kwargs)
         model_output = self.model.run(None, onnx_input)  # type: ignore[union-attr]
 
         return OnnxOutputContext(
             model_output=model_output[0],
-            attention_mask=attention_mask,
+            attention_mask=attention_mask,  # type: ignore[arg-type]
             metadata=metadata,
         )
 
@@ -231,7 +234,7 @@ def _process_nested_patches(
                 attention_mask[i, j] = 1
 
         metadata = {"patch_counts": patch_counts}
-        return encoded, attention_mask, metadata
+        return encoded, attention_mask, metadata  # type: ignore[return-value]
 
     def _process_flat_images(
         self, processed: list[NumpyArray], num_images: int
@@ -269,7 +272,7 @@ def _process_flat_images(
             attention_mask = np.ones((num_images, 1), dtype=np.int64)
             metadata = {"patch_counts": [1] * num_images}
 
-        return encoded, attention_mask, metadata
+        return encoded, attention_mask, metadata  # type: ignore[return-value]
 
     def _needs_patch_dimension(self) -> bool:
         """