Skip to content

Commit e462bf0

Browse files
mypy fixes
1 parent e111ab9 commit e462bf0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

fastembed/late_interaction_multimodal/colmodernvbert.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -165,8 +165,8 @@ def _preprocess_onnx_text_input(
165165
Iterable[NumpyArray]: Post-processed output as NumPy arrays.
166166
"""
167167
batch_size, seq_length = onnx_input["input_ids"].shape
168-
empty_image_placeholder: NumpyArray = np.zeros( # type: ignore[type-var,arg-type,assignment]
169-
(batch_size, seq_length, 3, self.image_size, self.image_size), dtype=np.float32
168+
empty_image_placeholder: NumpyArray = np.zeros(
169+
(batch_size, seq_length, 3, self.image_size, self.image_size), dtype=np.float32 # type: ignore[type-var,arg-type,assignment]
170170
)
171171
onnx_input["pixel_values"] = empty_image_placeholder
172172
return onnx_input

0 commit comments

Comments
 (0)