From f26d0ddbbdcdab76223899ccfef6b94692d44e9e Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Fri, 15 Jul 2022 23:45:47 +0400 Subject: [PATCH] Minor fix to asymmetric search when no entries returned --- src/search_type/asymmetric.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/search_type/asymmetric.py b/src/search_type/asymmetric.py index 47526393..f8a750f2 100644 --- a/src/search_type/asymmetric.py +++ b/src/search_type/asymmetric.py @@ -14,7 +14,6 @@ from sentence_transformers import SentenceTransformer, CrossEncoder, util # Internal Packages from src.utils.helpers import get_absolute_path, resolve_absolute_path, load_model from src.processor.org_mode.org_to_jsonl import org_to_jsonl -from src.search_filter.explicit_filter import explicit_filter from src.utils.config import TextSearchModel from src.utils.rawconfig import AsymmetricSearchConfig, TextContentConfig from src.utils.constants import empty_escape_sequences @@ -106,7 +105,7 @@ def query(raw_query: str, model: TextSearchModel, device=torch.device('cpu'), fi for filter in filters: query, entries, corpus_embeddings = filter(query, entries, corpus_embeddings) if entries is None or len(entries) == 0: - return {} + return [], [] # Encode the query using the bi-encoder question_embedding = model.bi_encoder.encode([query], convert_to_tensor=True)