update ui.plot when leaving its context

showmethemoney2022 · Oct 27, 2022 · 0d7392c · 0d7392c
1 parent 2450748
commit 0d7392c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/nicegui/elements/plot.py b/nicegui/elements/plot.py
@@ -24,11 +24,10 @@ def __init__(self, *, close: bool = True, **kwargs):
 
     def __enter__(self):
         plt.figure(self.fig)
-
         return self
 
     def __exit__(self, *_):
         self.view.set_figure(plt.gcf())
-
         if self.close:
             plt.close(self.fig)
+        self.update()