remove Union usage

kacperlukawski · kacperlukawski · commit 986c6ba639b1 · 2025-12-15T12:22:48.000+01:00
diff --git a/fastembed/late_interaction_multimodal/colpali.py b/fastembed/late_interaction_multimodal/colpali.py
@@ -301,7 +301,7 @@ def embed_image(
 
     def get_image_mask(
         self,
-        images: Union[ImageInput, Iterable[ImageInput]],
+        images: ImageInput | Iterable[ImageInput],
         batch_size: int = 16,
         **kwargs: Any,
     ) -> list[NumpyArray]:
diff --git a/fastembed/late_interaction_multimodal/late_interaction_multimodal_embedding.py b/fastembed/late_interaction_multimodal/late_interaction_multimodal_embedding.py
@@ -186,7 +186,7 @@ def token_count(
 
     def get_image_mask(
         self,
-        images: Union[ImageInput, Iterable[ImageInput]],
+        images: ImageInput | Iterable[ImageInput],
         batch_size: int = 16,
         **kwargs: Any,
     ) -> list[NumpyArray]:
diff --git a/fastembed/late_interaction_multimodal/late_interaction_multimodal_embedding_base.py b/fastembed/late_interaction_multimodal/late_interaction_multimodal_embedding_base.py
@@ -87,7 +87,7 @@ def token_count(
 
     def get_image_mask(
         self,
-        images: Union[ImageInput, Iterable[ImageInput]],
+        images: ImageInput | Iterable[ImageInput],
         batch_size: int = 16,
         **kwargs: Any,
     ) -> list[NumpyArray]:
