From b1ebe209d57cd5553eb058cf8ddf642d84a0238d Mon Sep 17 00:00:00 2001 From: jupyterjazz Date: Wed, 5 Jul 2023 15:16:46 +0400 Subject: [PATCH] fix: dont convert doclist Signed-off-by: jupyterjazz --- docarray/index/abstract.py | 4 ++-- docarray/index/backends/weaviate.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docarray/index/abstract.py b/docarray/index/abstract.py index 9b7f8d25513..740d054166f 100644 --- a/docarray/index/abstract.py +++ b/docarray/index/abstract.py @@ -578,7 +578,7 @@ def filter( self._logger.debug(f'Executing `filter` for the query {filter_query}') docs = self._filter(filter_query, limit=limit, **kwargs) - if isinstance(docs, List): + if isinstance(docs, List) and not isinstance(docs, DocList): docs = self._dict_list_to_docarray(docs) return docs @@ -656,7 +656,7 @@ def text_search( query_text, search_field=search_field, limit=limit, **kwargs ) - if isinstance(docs, List): + if isinstance(docs, List) and not isinstance(docs, DocList): docs = self._dict_list_to_docarray(docs) return FindResult(documents=docs, scores=scores) diff --git a/docarray/index/backends/weaviate.py b/docarray/index/backends/weaviate.py index c6354a94379..2eba589efa9 100644 --- a/docarray/index/backends/weaviate.py +++ b/docarray/index/backends/weaviate.py @@ -357,7 +357,7 @@ def find( query_vec_np, search_field=search_field, limit=limit, **kwargs ) - if isinstance(docs, List): + if isinstance(docs, List) and not isinstance(docs, DocList): docs = self._dict_list_to_docarray(docs) return FindResult(documents=docs, scores=scores)