diff --git a/tests/test_optional_dependencies.py b/tests/test_optional_dependencies.py index fd0d8a516..08b5c955e 100644 --- a/tests/test_optional_dependencies.py +++ b/tests/test_optional_dependencies.py @@ -1,8 +1,10 @@ import importlib import sys from unittest import mock + import pytest + def test_rag_storage_without_chromadb(): # Mock the import to simulate chromadb not being installed with mock.patch.dict(sys.modules, {'chromadb': None}):