Skip to content

Commit

Permalink
Handle UCXNotConnected error (#5449)
Browse files Browse the repository at this point in the history
Handle UCXNotConnected error
  • Loading branch information
pentschev authored Oct 21, 2021
1 parent 3afc670 commit 918e3fb
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions distributed/comm/ucx.py
Original file line number Diff line number Diff line change
Expand Up @@ -383,10 +383,10 @@ async def connect(self, address: str, deserialize=True, **connection_args) -> UC
init_once()
try:
ep = await ucx_create_endpoint(ip, port)
except (
ucp.exceptions.UCXCloseError,
ucp.exceptions.UCXCanceled,
) + (getattr(ucp.exceptions, "UCXConnectionReset", ()),):
except (ucp.exceptions.UCXCloseError, ucp.exceptions.UCXCanceled,) + (
getattr(ucp.exceptions, "UCXConnectionReset", ()),
getattr(ucp.exceptions, "UCXNotConnected", ()),
):
raise CommClosedError("Connection closed before handshake completed")
return self.comm_class(
ep,
Expand Down

0 comments on commit 918e3fb

Please sign in to comment.