diff --git a/tests/unit/encoders/test_clip.py b/tests/unit/encoders/test_clip.py index 2a9e814cbee7472ab3d93c6c8a49b78b77633e58..fc057e858ddaf601d35203b04ec64af1fa6f649a 100644 --- a/tests/unit/encoders/test_clip.py +++ b/tests/unit/encoders/test_clip.py @@ -1,7 +1,7 @@ import numpy as np import pytest -from PIL import Image import torch +from PIL import Image from semantic_router.encoders import CLIPEncoder diff --git a/tests/unit/encoders/test_vit.py b/tests/unit/encoders/test_vit.py index 01aa32883ad4086d12eca953cc6b91b214551958..0df7f4627609a8d7274c1da4a4ffcbde912e820c 100644 --- a/tests/unit/encoders/test_vit.py +++ b/tests/unit/encoders/test_vit.py @@ -1,7 +1,7 @@ import numpy as np import pytest -from PIL import Image import torch +from PIL import Image from semantic_router.encoders import VitEncoder