diff --git a/docarray/index/abstract.py b/docarray/index/abstract.py index 13f4837cd61..783b701581d 100644 --- a/docarray/index/abstract.py +++ b/docarray/index/abstract.py @@ -387,7 +387,8 @@ def index(self, docs: Union[BaseDoc, Sequence[BaseDoc]], **kwargs): :param docs: Documents to index. """ - self._logger.debug(f'Indexing {len(docs)} documents') + n_docs = 1 if isinstance(docs, BaseDoc) else len(docs) + self._logger.debug(f'Indexing {n_docs} documents') docs_validated = self._validate_docs(docs) data_by_columns = self._get_col_value_dict(docs_validated) self._index(data_by_columns, **kwargs) diff --git a/docarray/index/backends/hnswlib.py b/docarray/index/backends/hnswlib.py index 6caf5817272..756f80f78e4 100644 --- a/docarray/index/backends/hnswlib.py +++ b/docarray/index/backends/hnswlib.py @@ -203,7 +203,8 @@ def index(self, docs: Union[BaseDoc, Sequence[BaseDoc]], **kwargs): if kwargs: raise ValueError(f'{list(kwargs.keys())} are not valid keyword arguments') - self._logger.debug(f'Indexing {len(docs)} documents') + n_docs = 1 if isinstance(docs, BaseDoc) else len(docs) + self._logger.debug(f'Indexing {n_docs} documents') docs_validated = self._validate_docs(docs) data_by_columns = self._get_col_value_dict(docs_validated) hashed_ids = tuple(self._to_hashed_id(doc.id) for doc in docs_validated)