diff --git a/tests/unit/encoders/test_vit.py b/tests/unit/encoders/test_vit.py index f11809850b2700b0f84f838cf250331a38f33e5d..b598a818a3af0753a0627fead4801a56c5b03f54 100644 --- a/tests/unit/encoders/test_vit.py +++ b/tests/unit/encoders/test_vit.py @@ -40,9 +40,7 @@ class TestVitEncoder: with pytest.raises(ImportError) as error: VitEncoder() - assert "Please install transformers to use VitEncoder" in str( - error.value - ) + assert "Please install transformers to use VitEncoder" in str(error.value) @pytest.mark.skipif( os.environ.get("RUN_HF_TESTS") is None, reason="Set RUN_HF_TESTS=1 to run"