From 5e6e3b9237e38c82834f34fcd8ef7aadbe1bfe60 Mon Sep 17 00:00:00 2001 From: xjules Date: Fri, 6 Dec 2024 14:47:44 +0100 Subject: [PATCH] Mypy fix --- src/_ert/forward_model_runner/client.py | 2 +- src/ert/ensemble_evaluator/_ensemble.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/_ert/forward_model_runner/client.py b/src/_ert/forward_model_runner/client.py index 1b92e9c938e..ae40415fbea 100644 --- a/src/_ert/forward_model_runner/client.py +++ b/src/_ert/forward_model_runner/client.py @@ -51,7 +51,7 @@ async def __aexit__( await self._term_receiver_task() self.term() - async def _term_receiver_task(self): + async def _term_receiver_task(self) -> None: if self._receiver_task and not self._receiver_task.done(): self._receiver_task.cancel() await asyncio.gather(self._receiver_task, return_exceptions=True) diff --git a/src/ert/ensemble_evaluator/_ensemble.py b/src/ert/ensemble_evaluator/_ensemble.py index 3e300c3147d..c6b2f8a4146 100644 --- a/src/ert/ensemble_evaluator/_ensemble.py +++ b/src/ert/ensemble_evaluator/_ensemble.py @@ -203,8 +203,8 @@ async def send_event( cert: Optional[Union[str, bytes]] = None, retries: int = 10, ) -> None: - async with Client(url, token, cert, max_retries=retries) as client: - await client._send(event_to_json(event)) + async with Client(url, token, cert) as client: + await client._send(event_to_json(event), max_retries=retries) def generate_event_creator(self) -> Callable[[Id.ENSEMBLE_TYPES], Event]: def event_builder(status: str) -> Event: