From 6be3141329b60586dc6d380e5d2375f88c21c5c3 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Tue, 20 Feb 2024 01:25:09 +0400
Subject: [PATCH] lint
---
semantic_router/encoders/clip.py | 4 ++--
semantic_router/encoders/vit.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/semantic_router/encoders/clip.py b/semantic_router/encoders/clip.py
index c1b87779..c7210707 100644
--- a/semantic_router/encoders/clip.py
+++ b/semantic_router/encoders/clip.py
@@ -66,7 +66,7 @@ class CLIPEncoder(BaseEncoder):
"You can install it with: "
"`pip install semantic-router[vision]`"
)
-
+
try:
from PIL import Image
except ImportError:
@@ -114,7 +114,7 @@ class CLIPEncoder(BaseEncoder):
embeds = self._model.get_image_features(pixel_values=inputs)
embeds = embeds.squeeze(0).cpu().detach().numpy()
return embeds
-
+
def _ensure_rgb(self, img: Any):
rgbimg = self._Image.new("RGB", img.size)
rgbimg.paste(img)
diff --git a/semantic_router/encoders/vit.py b/semantic_router/encoders/vit.py
index 9aba6257..801d2994 100644
--- a/semantic_router/encoders/vit.py
+++ b/semantic_router/encoders/vit.py
@@ -41,7 +41,7 @@ class VitEncoder(BaseEncoder):
"You can install it with: "
"`pip install semantic-router[vision]`"
)
-
+
try:
from PIL import Image
except ImportError:
--
GitLab