diff --git a/coworker/connectors/relay_client.py b/coworker/connectors/relay_client.py index 57fc38cc..f1e7bf1c 100644 --- a/coworker/connectors/relay_client.py +++ b/coworker/connectors/relay_client.py @@ -204,10 +204,10 @@ async def wait_dispatched(self, at_least: int, timeout: float = 2.0) -> None: ) try: await asyncio.wait_for(self._progress.wait(), timeout=remaining) - except asyncio.TimeoutError: + except asyncio.TimeoutError as exc: raise TimeoutError( f"only {self._dispatched} frames dispatched (< {at_least})" - ) + ) from exc # -- default transport --------------------------------------------------- def _default_transport_factory(self) -> RelayTransport: