refactor: added logic to pass postprocess if it's not a template that…

… needs processing
brown-ccv · Sep 19, 2024 · 174e451 · 174e451
1 parent e7d03f0
commit 174e451
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/hnn_core/gui/_viz_manager.py b/hnn_core/gui/_viz_manager.py
@@ -776,6 +776,9 @@ def _postprocess_template(template_name, fig, idx,
     this function. For example, L2 and L5 dipole plots should have the same
     y-axis range.
     """
+    if template_name not in ['Dipole Layers (3x1)']:
+        return
+
     if template_name == 'Dipole Layers (3x1)':
         # Make the L2 and L5 plots the same y-range
         y_lims_list = [ax.get_ylim() for ax in fig.axes[0:2]]