From b68729cee614bff33da58c672cf1ef7072b75eda Mon Sep 17 00:00:00 2001
From: George Dang <53052793+gtdang@users.noreply.github.com>
Date: Thu, 19 Sep 2024 16:44:51 -0400
Subject: [PATCH] refactor: added logic to pass postprocess if it's not a
 template that needs processing

---
 hnn_core/gui/_viz_manager.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/hnn_core/gui/_viz_manager.py b/hnn_core/gui/_viz_manager.py
index db3f8c7b2..a872ecd64 100644
--- 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]]