upstream-archive/fastembed
1
0
Fork 0
mirror of https://github.com/qdrant/fastembed.git synced 2026-10-03 03:17:48 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix mypy errors

Browse Source
This commit is contained in:
Kacper Łukawski
2026-01-02 21:02:38 +01:00
parent 96eb50bcbc
commit cf4100c9de
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastembed/late_interaction_multimodal/onnx_multimodal_model.py
+7 -4
View File
@@ -184,7 +184,10 @@ class OnnxMultimodalModel(OnnxModel[T]):
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)
@@ -192,7 +195,7 @@ class OnnxMultimodalModel(OnnxModel[T]):
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 @@ class OnnxMultimodalModel(OnnxModel[T]):
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 @@ class OnnxMultimodalModel(OnnxModel[T]):
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:
"""