refactor: move figure header and footer commands into context manager

brown-ccv · Sep 27, 2024 · 1e87fc1 · 1e87fc1
1 parent 5aa7d1c
commit 1e87fc1
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/hnn_core/gui/_viz_manager.py b/hnn_core/gui/_viz_manager.py
@@ -763,9 +763,8 @@ def _add_figure(b, widgets, data, template_type, scale=0.95, dpi=96):
         with plt.ioff():
             fig = plt.figure(figsize=figsize, dpi=dpi, layout='constrained')
             axd = fig.subplot_mosaic(mosaic, **kwargs)
-
-        fig.canvas.header_visible = False
-        fig.canvas.footer_visible = False
+            fig.canvas.header_visible = False
+            fig.canvas.footer_visible = False
 
         if data['use_ipympl'] is False:
             plt.show()