Fix mistake in tests

deepsense-ai · Nov 13, 2024 · 995848f · 995848f
1 parent 4a38a82
commit 995848f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/packages/ragbits-document-search/tests/unit/test_document_search.py b/packages/ragbits-document-search/tests/unit/test_document_search.py
@@ -11,7 +11,7 @@
 from ragbits.document_search.documents.element import TextElement
 from ragbits.document_search.documents.sources import LocalFileSource
 from ragbits.document_search.ingestion.document_processor import DocumentProcessorRouter
-from ragbits.document_search.ingestion.processor_strategies.sequential import SequentialProcessing
+from ragbits.document_search.ingestion.processor_strategies.batched import BatchedAsyncProcessing
 from ragbits.document_search.ingestion.providers import BaseProvider
 from ragbits.document_search.ingestion.providers.dummy import DummyProvider
 
@@ -186,8 +186,7 @@ async def test_document_search_with_batched():
     embeddings_mock = AsyncMock()
     embeddings_mock.embed_text.return_value = [[0.1, 0.1]] * len(documents)
 
-    # processing_strategy = BatchedAsyncProcessing(batch_size=5)
-    processing_strategy = SequentialProcessing()
+    processing_strategy = BatchedAsyncProcessing(batch_size=5)
     vectore_store = InMemoryVectorStore()
 
     document_search = DocumentSearch(