Add missing tree_pop() in autogui

joruof · Apr 29, 2024 · 1a33b9f · 1a33b9f
1 parent 4ae13c6
commit 1a33b9f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/imviz/autogui.py b/imviz/autogui.py
@@ -266,6 +266,8 @@ def render(self, obj, name=""):
         else:
             if len(name) == 0:
                 viz.text(f"{name}: " + "???")
+            if tree_open:
+                viz.tree_pop()
             return obj
 
         obj_annots = getattr(obj, "__annotations__", {})