upd

YieldsLabs · Sep 14, 2024 · d8129ee · d8129ee
1 parent 9f5b578
commit d8129ee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/exchange/_bybit_ws.py b/exchange/_bybit_ws.py
@@ -65,7 +65,7 @@ async def _connect(self):
                 logger.info("WebSocket connection established.")
             except Exception as e:
                 logger.error(f"Failed to connect to WebSocket: {e}")
-                raise ConnectionError(e)
+                raise ConnectionError("Failed to connect to WebSocket") from None
 
     @retry(
         max_retries=13,
@@ -150,7 +150,7 @@ async def _wait_for_ws(self, timeout=10):
             await asyncio.wait_for(self._check_ws_open(), timeout=timeout)
         except asyncio.TimeoutError:
             logger.error("Timed out waiting for WebSocket to open.")
-            raise ConnectionError("WebSocket connection timeout.")
+            raise ConnectionError("WebSocket connection timeout.") from None
 
     async def _check_ws_open(self):
         while not self.ws or not self.ws.open: