diff --git a/docarray/index/abstract.py b/docarray/index/abstract.py index 3c423137259..774ed71b7c8 100644 --- a/docarray/index/abstract.py +++ b/docarray/index/abstract.py @@ -573,7 +573,8 @@ def text_search_batched( docs = [self._dict_list_to_docarray(docs) for docs in da_list] return FindResultBatched(documents=docs, scores=scores) - return FindResultBatched(documents=da_list, scores=scores) + da_list_ = cast(List[DocList], da_list) + return FindResultBatched(documents=da_list_, scores=scores) ########################################################## # Helper methods # diff --git a/docarray/index/backends/hnswlib.py b/docarray/index/backends/hnswlib.py index d0e11e7e959..2807ba9b06b 100644 --- a/docarray/index/backends/hnswlib.py +++ b/docarray/index/backends/hnswlib.py @@ -29,6 +29,7 @@ _raise_not_supported, ) from docarray.proto import DocProto +from docarray.typing import NdArray from docarray.typing.tensor.abstract_tensor import AbstractTensor from docarray.utils._internal.misc import import_library, is_np_int from docarray.utils.filter import filter_docs @@ -262,7 +263,9 @@ def _find( docs, scores = self._find_batched( queries=query_batched, limit=limit, search_field=search_field ) - return _FindResult(documents=docs[0], scores=scores[0]) + return _FindResult( + documents=docs[0], scores=NdArray._docarray_from_native(scores[0]) + ) def _filter( self,