diff --git a/docs/ert/conf.py b/docs/ert/conf.py index fa79b90b611..a16d77b86cb 100644 --- a/docs/ert/conf.py +++ b/docs/ert/conf.py @@ -68,7 +68,6 @@ ("py:class", "pydantic.types.PositiveInt"), ("py:class", "LibresFacade"), ("py:class", "pandas.core.frame.DataFrame"), - ("py:class", "websockets.server.WebSocketServerProtocol"), ("py:class", "EnsembleReader"), ] nitpick_ignore_regex = [ diff --git a/src/ert/ensemble_evaluator/evaluator.py b/src/ert/ensemble_evaluator/evaluator.py index 522ecaadc7d..1af2c50003b 100644 --- a/src/ert/ensemble_evaluator/evaluator.py +++ b/src/ert/ensemble_evaluator/evaluator.py @@ -269,8 +269,6 @@ async def listen_for_messages(self) -> None: await self.forward_checksum(event) else: await self._events.put(event) - # if type(event) in [EnsembleSucceeded, EnsembleFailed]: - # return else: logger.info(f"Connection attempt to unknown sender: {sender}.") except zmq.error.ZMQError as e: diff --git a/src/ert/ensemble_evaluator/monitor.py b/src/ert/ensemble_evaluator/monitor.py index 2c980fb17ab..3d999b471b9 100644 --- a/src/ert/ensemble_evaluator/monitor.py +++ b/src/ert/ensemble_evaluator/monitor.py @@ -139,7 +139,6 @@ async def _receiver(self) -> None: except zmq.ZMQError as exc: logger.error(f"Failed to connect to {self._ee_con_info.router_uri} {exc}") raise - await self._socket.send_multipart([b"", b"CONNECT"]) self._connected.set() diff --git a/tests/ert/unit_tests/ensemble_evaluator/test_ensemble_legacy.py b/tests/ert/unit_tests/ensemble_evaluator/test_ensemble_legacy.py index 3a8935b26a8..07f0f55fb8b 100644 --- a/tests/ert/unit_tests/ensemble_evaluator/test_ensemble_legacy.py +++ b/tests/ert/unit_tests/ensemble_evaluator/test_ensemble_legacy.py @@ -46,7 +46,7 @@ async def test_run_legacy_ensemble( ) async with ( evaluator_to_use(ensemble, config) as evaluator, - Monitor(config) as monitor, + Monitor(config.get_connection_info()) as monitor, ): async for event in monitor.track(): if type(event) in ( @@ -85,7 +85,7 @@ async def test_run_and_cancel_legacy_ensemble( async with ( evaluator_to_use(ensemble, config) as evaluator, - Monitor(config) as monitor, + Monitor(config.get_connection_info()) as monitor, ): # on lesser hardware the realizations might be killed by max_runtime # and the ensemble is set to STOPPED