trying to catch and restart websocket closed exceptions

coinmetrics · Dec 6, 2024 · 2d616c3 · 2d616c3
1 parent 781080f
commit 2d616c3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/precog/validators/validator.py b/precog/validators/validator.py
@@ -2,6 +2,7 @@
 from pathlib import Path
 
 import bittensor as bt
+import websocket
 
 from precog.utils.classes import Config
 from precog.utils.general import parse_arguments
@@ -27,6 +28,9 @@ async def main(self):
         except BrokenPipeError:
             bt.logging.error("Recieved a Broken Pipe substrate error")
             asyncio.run(self.reset_instance())
+        except websocket._exceptions.WebSocketConnectionClosedException:
+            bt.logging.error("Recieved a websocket closed error, restarting validator")
+            asyncio.run(self.reset_instance())
         except Exception as e:
             bt.logging.error(f"Unhandled exception: {e}")
         finally: