Skip to content

Commit 9b0f4cb

Browse files
Fix mypy errors
1 parent 091f697 commit 9b0f4cb

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

fastembed/late_interaction_multimodal/onnx_multimodal_model.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,8 @@ def onnx_embed_image(self, images: list[ImageInput], **kwargs: Any) -> OnnxOutpu
185185
encoded, attention_mask, metadata = self._process_nested_patches(processed)
186186
else:
187187
encoded, attention_mask, metadata = self._process_flat_images(
188-
processed, len(images) # type: ignore[arg-type]
188+
processed, # type: ignore[arg-type]
189+
len(images),
189190
)
190191

191192
onnx_input = {"pixel_values": encoded, "attention_mask": attention_mask}
@@ -233,7 +234,7 @@ def _process_nested_patches(
233234
attention_mask[i, j] = 1
234235

235236
metadata = {"patch_counts": patch_counts}
236-
return encoded, attention_mask, metadata
237+
return encoded, attention_mask, metadata # type: ignore[return-value]
237238

238239
def _process_flat_images(
239240
self, processed: list[NumpyArray], num_images: int

0 commit comments

Comments
 (0)