Change errors to ConnectionErrors in fw.client

xjules · Dec 1, 2024 · 552e5d1 · 552e5d1
1 parent d8d8c9d
commit 552e5d1
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/_ert/forward_model_runner/client.py b/src/_ert/forward_model_runner/client.py
@@ -95,10 +95,12 @@ async def reconnect(self) -> None:
                 self.socket.recv_multipart(), timeout=self._connection_timeout
             )
             if ack.decode() != "ACK":
-                raise asyncio.TimeoutError("No Ack for connect")
-        except asyncio.TimeoutError:
+                raise ClientConnectionError("No Ack for connect")
+        except asyncio.TimeoutError as exc:
             logger.warning("Failed to get acknowledgment on dealer connect!")
-            raise
+            raise ClientConnectionError(
+                "Connection to evaluator not established!"
+            ) from exc
 
     def send(self, messages: str | list[str]) -> None:
         self.loop.run_until_complete(self._send(messages))