From 905492a3db6ebb52a22c96123a499264b2d4b904 Mon Sep 17 00:00:00 2001 From: niklas Date: Mon, 30 Oct 2023 10:25:43 +0100 Subject: [PATCH] Fix retriever fixture names --- tests/conftest.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index 9ebf20349..e811937db 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -45,24 +45,24 @@ def prompt_image() -> Image: @fixture -def asymmetric_qdrant_in_memory_retriever( - client: Client, qdrant_in_memory_retriever_documents: Sequence[Document] +def asymmetric_in_memory_retriever( + client: Client, in_memory_retriever_documents: Sequence[Document] ) -> QdrantInMemoryRetriever: return QdrantInMemoryRetriever( client, - qdrant_in_memory_retriever_documents, + in_memory_retriever_documents, k=2, retriever_type=RetrieverType.ASYMMETRIC, ) @fixture -def symmetric_qdrant_in_memory_retriever( - client: Client, qdrant_in_memory_retriever_documents: Sequence[Document] +def symmetric_in_memory_retriever( + client: Client, in_memory_retriever_documents: Sequence[Document] ) -> QdrantInMemoryRetriever: return QdrantInMemoryRetriever( client, - qdrant_in_memory_retriever_documents, + in_memory_retriever_documents, k=2, retriever_type=RetrieverType.SYMMETRIC, )