Add wait for ndif connection

ndif-team · Dec 28, 2023 · 1126bc9 · 1126bc9
1 parent 20ef1fe
commit 1126bc9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/nnsight/contexts/Runner.py b/src/nnsight/contexts/Runner.py
@@ -95,12 +95,13 @@ def run_server(self):
 
     def blocking_request(self, request: pydantics.RequestModel):
         # Create a socketio connection to the server.
-        sio = socketio.Client(logger=logger)
+        sio = socketio.Client(logger=logger, reconnection_attempts=10)
 
         sio.connect(
             f"wss://{CONFIG.API.HOST}",
             socketio_path="/ws/socket.io",
             transports=["websocket"],
+            wait_timeout=10
         )
 
         # Called when receiving a response from the server.