diff --git a/pandagg/index.py b/pandagg/index.py index 1d516c66..1e1b6235 100644 --- a/pandagg/index.py +++ b/pandagg/index.py @@ -269,7 +269,7 @@ def to_dict(self) -> Dict[str, Any]: d["version"] = self.version return d - def save(self) -> None: + def save(self) -> Any: return self._get_connection().indices.put_index_template( name=self.name, body=self.to_dict() ) diff --git a/pandagg/search.py b/pandagg/search.py index 381d7682..6ac575dd 100644 --- a/pandagg/search.py +++ b/pandagg/search.py @@ -792,7 +792,7 @@ def execute(self) -> SearchResponse: """ es = self._get_connection() raw_data = es.search(index=self._index, body=self.to_dict()) - return SearchResponse(data=raw_data, _search=self) + return SearchResponse(data=raw_data, _search=self) # type: ignore def scan_composite_agg(self, size: int) -> Iterator[BucketDict]: """Iterate over the whole aggregation composed buckets, yields buckets.""" @@ -851,7 +851,7 @@ def delete(self) -> DeleteByQueryResponse: """ es = self._get_connection() - return es.delete_by_query(index=self._index, body=self.to_dict()) + return es.delete_by_query(index=self._index, body=self.to_dict()) # type: ignore def __eq__(self, other: Any) -> bool_: return ( @@ -942,7 +942,7 @@ def execute(self) -> List[SearchResponseDict]: Execute the multi search request and return a list of search results. """ es = self._get_connection() - return es.msearch(index=self._index, body=self.to_dict(), **self._params) + return es.msearch(index=self._index, body=self.to_dict(), **self._params) # type: ignore def __eq__(self, other: Any) -> bool: return (