refactor: Asyncio timeoutError update

8 months ago · 79f0271720
parent 5b8b80cf59
commit 79f0271720
1 changed files with 2 additions and 2 deletions
--- a/scripts/ws_client.py
+++ b/scripts/ws_client.py
@ -234,7 +234,7 @@ async def create_client(

                        await protocol.on_message(ws, name, message)

-                    except (asyncio.TimeoutError, websockets.exceptions.WebSocketException):
+                    except (TimeoutError, websockets.exceptions.WebSocketException):
                        # Try pinging
                        try:
                            pong = await ws.ping()
@ -244,7 +244,7 @@ async def create_client(

                            continue

-                        except asyncio.TimeoutError:
+                        except TimeoutError:
                            logger.error(f"Ping timed out, retrying in {sleep_time}s")
                            await asyncio.sleep(sleep_time)