Quick fix to ZMQ deadlock error (#784)

quick fix
nerfstudio-project · Oct 16, 2022 · 555cec9 · 555cec9
1 parent d4670de
commit 555cec9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/nerfstudio/viewer/server/viewer_utils.py b/nerfstudio/viewer/server/viewer_utils.py
@@ -236,9 +236,11 @@ def __init__(self, config: cfg.ViewerConfig, log_filename: Path):
             CONSOLE.rule(characters="=")
             CONSOLE.line()
             self.vis = Viewer(zmq_port=zmq_port)
+            self.vis_webrtc_thread = Viewer(zmq_port=zmq_port)
         else:
             assert self.config.zmq_port is not None
             self.vis = Viewer(zmq_port=self.config.zmq_port, ip_address=self.config.ip_address)
+            self.vis_webrtc_thread = Viewer(zmq_port=self.config.zmq_port, ip_address=self.config.ip_address)
 
         # viewer specific variables
         self.prev_camera_matrix = None
@@ -488,8 +490,10 @@ async def send_webrtc_answer(self, data):
         answer = await pc.createAnswer()
         await pc.setLocalDescription(answer)
 
-        self.vis["webrtc/answer"].write({"sdp": pc.localDescription.sdp, "type": pc.localDescription.type})
-        self.vis["webrtc/answer"].delete()
+        self.vis_webrtc_thread["webrtc/answer"].write(
+            {"sdp": pc.localDescription.sdp, "type": pc.localDescription.type}
+        )
+        self.vis_webrtc_thread["webrtc/answer"].delete()
 
         # continually exchange media
         while True: