fix: Fix text and refactor hybrid search

superlinear-ai · Dec 13, 2024 · 8e6436e · 8e6436e
1 parent 2a14928
commit 8e6436e
Showing 1 changed file with 0 additions and 7 deletions.
diff --git a/tests/test_rerank.py b/tests/test_rerank.py
@@ -1,7 +1,6 @@
 """Test RAGLite's reranking functionality."""
 
 import random
-from functools import partial
 from typing import TypeVar
 
 import pytest
@@ -12,10 +11,8 @@
 from raglite import (
     RAGLiteConfig,
     hybrid_search,
-    keyword_search,
     rerank_chunks,
     retrieve_chunks,
-    vector_search,
 )
 from raglite._database import Chunk
 
@@ -65,10 +62,6 @@ def test_reranker(
     chunk_ids, _ = hybrid_search(
         query,
         config=raglite_test_config,
-        subsearches=[
-            partial(vector_search, max_chunks=40, config=raglite_test_config),
-            partial(keyword_search, max_chunks=40, config=raglite_test_config),
-        ],
         max_chunks=20,
     )
     # Retrieve the chunks.