diff --git a/src/ert/ensemble_evaluator/_ensemble.py b/src/ert/ensemble_evaluator/_ensemble.py index f752ded6601..e682563dc9f 100644 --- a/src/ert/ensemble_evaluator/_ensemble.py +++ b/src/ert/ensemble_evaluator/_ensemble.py @@ -31,7 +31,6 @@ from ert.run_arg import RunArg from ert.scheduler import Scheduler, create_driver -from ._wait_for_evaluator import wait_for_evaluator from .config import EvaluatorServerConfig from .snapshot import EnsembleSnapshot, FMStepSnapshot, RealizationSnapshot from .state import ( @@ -200,8 +199,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)) + with Client(url, token, cert, max_retries=retries) as client: + client.send(event_to_json(event)) def generate_event_creator(self) -> Callable[[Id.ENSEMBLE_TYPES], Event]: def event_builder(status: str) -> Event: @@ -226,16 +225,16 @@ async def evaluate( ce_unary_send_method_name, partialmethod( self.__class__.send_event, - self._config.dispatch_uri, + self._config.get_connection_info().push_pull_uri, token=self._config.token, cert=self._config.cert, ), ) - await wait_for_evaluator( - base_url=self._config.url, - token=self._config.token, - cert=self._config.cert, - ) + # await wait_for_evaluator( + # base_url=self._config.url, + # token=self._config.token, + # cert=self._config.cert, + # ) await self._evaluate_inner( event_unary_send=getattr(self, ce_unary_send_method_name), scheduler_queue=scheduler_queue,