Fix button remove in Gaussian test script

nerfstudio-project · Aug 5, 2024 · ba17c7c · ba17c7c
1 parent e43b4ac
commit ba17c7c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/experimental/gaussian_splats.py b/examples/experimental/gaussian_splats.py
@@ -140,7 +140,7 @@ def _(event: viser.GuiEvent) -> None:
         remove_button = server.gui.add_button(f"Remove splat object {i}")
 
         @remove_button.on_click
-        def _(_, gs_handle=gs_handle) -> None:
+        def _(_, gs_handle=gs_handle, remove_button=remove_button) -> None:
             gs_handle.remove()
             remove_button.remove()