docarray · AnneYang720 · Mar 29, 2023 · Mar 1, 2023 · Mar 2, 2023 · Mar 3, 2023
diff --git a/docarray/index/__init__.py b/docarray/index/__init__.py
@@ -1,3 +1,4 @@
+from docarray.index.backends.elastic import ElasticV7DocIndex
 from docarray.index.backends.hnswlib import HnswDocumentIndex
 
-__all__ = ['HnswDocumentIndex']
+__all__ = ['HnswDocumentIndex', 'ElasticV7DocIndex']
diff --git a/docarray/index/abstract.py b/docarray/index/abstract.py
@@ -9,6 +9,7 @@
     Generic,
     Iterable,
     List,
+    Mapping,
     NamedTuple,
     Optional,
     Sequence,
@@ -233,13 +234,13 @@ def _find(
     @abstractmethod
     def _find_batched(
         self,
-        query: np.ndarray,
+        queries: np.ndarray,
         limit: int,
         search_field: str = '',
     ) -> _FindResultBatched:
         """Find documents in the index
 
-        :param query: query vectors for KNN/ANN search.
+        :param queries: query vectors for KNN/ANN search.
             Has shape (batch_size, vector_dim)
         :param limit: maximum number of documents to return
         :param search_field: name of the field to search on
@@ -593,7 +594,7 @@ def _get_values_by_column(docs: Sequence[BaseDoc], col_name: str) -> List[Any]:
 
     @staticmethod
     def _transpose_col_value_dict(
-        col_value_dict: Dict[str, Iterable[Any]]
+        col_value_dict: Mapping[str, Iterable[Any]]
     ) -> Generator[Dict[str, Any], None, None]:
         """'Transpose' the output of `_get_col_value_dict()`: Yield rows of columns, where each row represent one Document.
         Since a generator is returned, this process comes at negligible cost.