diff --git a/osbenchmark/client.py b/osbenchmark/client.py index e1d71001..90bfa92e 100644 --- a/osbenchmark/client.py +++ b/osbenchmark/client.py @@ -363,11 +363,11 @@ async def on_request_end(session, trace_config_ctx, params): class BenchmarkAsyncOpenSearch(opensearchpy.AsyncOpenSearch, RequestContextHolder): print("benchmark###################") - async def __aenter__(self) -> Any: + async def __aenter__(self): print("enter###############") return self - async def __aexit__(self, *_: Any) -> None: + async def __aexit__(self, *_): print("exit###############") await self.close() #pass