fix: remove Union references

Author: kacperlukawski

fastembed/image/transform/functional.py

@@ -150,7 +150,7 @@ def pad2square(
 def resize_longest_edge(
     image: Image.Image,
     max_size: int,
-    resample: Union[int, Image.Resampling] = Image.Resampling.LANCZOS,
+    resample: int | Image.Resampling = Image.Resampling.LANCZOS,
 ) -> Image.Image:
     height, width = image.height, image.width
     aspect_ratio = width / height
@@ -192,7 +192,7 @@ def crop_ndarray(
 def resize_ndarray(
     image: NumpyArray,
     size: tuple[int, int],
-    resample: Union[int, Image.Resampling] = Image.Resampling.LANCZOS,
+    resample: int | Image.Resampling = Image.Resampling.LANCZOS,
     channel_first: bool = True,
 ) -> NumpyArray:
     # Convert to PIL-friendly format (H, W, C)

fastembed/image/transform/operators.py

@@ -42,8 +42,8 @@ def __init__(self, mean: float | list[float], std: float | list[float]):
         self.std = std
 
     def __call__(  # type: ignore[override]
-        self, images: Union[list[NumpyArray], list[list[NumpyArray]]]
-    ) -> Union[list[NumpyArray], list[list[NumpyArray]]]:
+        self, images: list[NumpyArray] | list[list[NumpyArray]]
+    ) -> list[NumpyArray] | list[list[NumpyArray]]:
         if images and isinstance(images[0], list):
             # Nested structure from ImageSplitter
             return [
@@ -73,8 +73,8 @@ def __init__(self, scale: float = 1 / 255):
         self.scale = scale
 
     def __call__(  # type: ignore[override]
-        self, images: Union[list[NumpyArray], list[list[NumpyArray]]]
-    ) -> Union[list[NumpyArray], list[list[NumpyArray]]]:
+        self, images: list[NumpyArray] | list[list[NumpyArray]]
+    ) -> list[NumpyArray] | list[list[NumpyArray]]:
         if images and isinstance(images[0], list):
             # Nested structure from ImageSplitter
             return [