Fixed multimodal interface types so that they work with 3.9

Author: x-tabdeveloping

turftopic/encoders/multimodal.py

@@ -1,8 +1,6 @@
 from typing import Protocol
 
 from PIL import Image
-from torch import Tensor
-from torch.utils.data import DataLoader
 
 
 class MultimodalEncoder(Protocol):
@@ -14,20 +12,20 @@ def get_text_embeddings(
         *,
         batch_size: int = 8,
         **kwargs,
-    ) -> Tensor: ...
+    ): ...
 
     def get_image_embeddings(
         self,
-        images: list[Image.Image] | DataLoader,
+        images: list[Image.Image],
         *,
         batch_size: int = 8,
         **kwargs,
-    ) -> Tensor: ...
+    ): ...
 
     def get_fused_embeddings(
         self,
         texts: list[str] = None,
         images: list[Image.Image] = None,
         batch_size: int = 8,
         **kwargs,
-    ) -> Tensor: ...
+    ): ...

turftopic/multimodal.py

@@ -1,14 +1,14 @@
 from __future__ import annotations
 
 from abc import abstractmethod
-from typing import Iterable, Optional, TypedDict
+from typing import Iterable, Optional, TypedDict, Union
 
 import numpy as np
 from PIL import Image
 
 UrlStr = str
 
-ImageRepr = [Image.Image | UrlStr]
+ImageRepr = Union[Image.Image, UrlStr]
 
 
 def _load_images(images: Iterable[ImageRepr]) -> Iterable[Image]: