Cleanup

nerfstudio-project · Jan 14, 2024 · 2b656ad · 2b656ad
1 parent 0fa9aa3
commit 2b656ad
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/examples/08_smplx_visualizer.py b/examples/08_smplx_visualizer.py
@@ -66,7 +66,9 @@ def main(
         full_pose = torch.from_numpy(
             onp.array(
                 [j.value for j in gui_elements.gui_joints[1:]], dtype=onp.float32
-            )[None, ...]  # type: ignore
+            )[
+                None, ...
+            ]  # type: ignore
         )
 
         # Get deformed mesh.

diff --git a/src/viser/_tunnel.py b/src/viser/_tunnel.py
@@ -270,7 +270,7 @@ async def relay(r: asyncio.StreamReader, w: asyncio.StreamWriter) -> None:
                         asyncio.create_task(relay(local_r, remote_w)),
                         asyncio.create_task(relay(remote_r, local_w)),
                     ),
-                    close_event.wait(),
+                    asyncio.create_task(close_event.wait()),
                 ],
                 return_when=asyncio.FIRST_COMPLETED,
             )

diff --git a/src/viser/client/src/ControlPanel/ControlPanel.tsx b/src/viser/client/src/ControlPanel/ControlPanel.tsx
@@ -59,10 +59,10 @@ export default function ControlPanel(props: {
     controlWidthString == "small"
       ? "16em"
       : controlWidthString == "medium"
-      ? "20em"
-      : controlWidthString == "large"
-      ? "24em"
-      : null
+        ? "20em"
+        : controlWidthString == "large"
+          ? "24em"
+          : null
   )!;
 
   const generatedServerToggleButton = (
@@ -201,6 +201,9 @@ function ShareButton() {
       setDoingSomething(false);
     }
   }, [shareUrl]);
+  React.useEffect(() => {
+    if (!connected && shareModalOpened) closeShareModal();
+  }, [connected, shareModalOpened]);
 
   if (viewer.useGui((state) => state.theme).show_share_button === false)
     return null;