From 555cec938784002f1d59fcb399e6fdacaaef18f2 Mon Sep 17 00:00:00 2001 From: Ethan Weber Date: Sun, 16 Oct 2022 11:39:11 -0700 Subject: [PATCH] Quick fix to ZMQ deadlock error (#784) quick fix --- nerfstudio/viewer/server/viewer_utils.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/nerfstudio/viewer/server/viewer_utils.py b/nerfstudio/viewer/server/viewer_utils.py index 1017ebf2b..791099ec4 100644 --- a/nerfstudio/viewer/server/viewer_utils.py +++ b/nerfstudio/viewer/server/viewer_utils.py @@ -236,9 +236,11 @@ class ViewerState: 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 @@ class ViewerState: 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: