From 995848fa39bf039dc30f8b82576a1d396f9495be Mon Sep 17 00:00:00 2001 From: Ludwik Trammer Date: Wed, 13 Nov 2024 18:55:25 +0100 Subject: [PATCH] Fix mistake in tests --- .../tests/unit/test_document_search.py | 5 ++--- 1 file changed, 2 insertions(+), 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 index 51bc0d97..b544ca47 100644 --- 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(