fix: tests

aurelio-labs · Aug 23, 2024 · 7f2cfae · 7f2cfae
1 parent 5cdf1d2
commit 7f2cfae
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/tests/unit/encoders/test_vit.py b/tests/unit/encoders/test_vit.py
@@ -43,11 +43,6 @@ def test_vit_encoder__import_errors_torch(self, mocker):
         with pytest.raises(ImportError):
             VitEncoder()
 
-    def test_vit_encoder__import_errors_torchvision(self, mocker):
-        mocker.patch.dict("sys.modules", {"torchvision": None})
-        with pytest.raises(ImportError):
-            VitEncoder()
-
     @pytest.mark.skipif(
         os.environ.get("RUN_HF_TESTS") is None, reason="Set RUN_HF_TESTS=1 to run"
     )