Fix error when we have too little data to have a cutoff

multiscale · Sep 1, 2023 · f0e4d00 · f0e4d00
1 parent 3c920f5
commit f0e4d00
Showing 1 changed file with 13 additions and 12 deletions.
diff --git a/muscle3/profiling.py b/muscle3/profiling.py
@@ -297,18 +297,19 @@ def update_data(self, ax: Axes) -> None:
                 for i in range(n_cur, n_avail):
                     bars[i].set_visible(False)
 
-            # update cutoff bars
-            bars = self._bars['_CUTOFF'].patches
-            if cutoff:
-                for bar in bars:
-                    bar.set_x(cutoff)
-                    bar.set_width(self._global_xmax - cutoff)
-                    bar.set_visible(True)
-                self._cutoff_warning.set_visible(True)
-            else:
-                for bar in bars:
-                    bar.set_visible(False)
-                self._cutoff_warning.set_visible(False)
+            # update cutoff bars, if any
+            if '_CUTOFF' in self._bars:
+                bars = self._bars['_CUTOFF'].patches
+                if cutoff:
+                    for bar in bars:
+                        bar.set_x(cutoff)
+                        bar.set_width(self._global_xmax - cutoff)
+                        bar.set_visible(True)
+                    self._cutoff_warning.set_visible(True)
+                else:
+                    for bar in bars:
+                        bar.set_visible(False)
+                    self._cutoff_warning.set_visible(False)
 
 
 tplot = None    # type: Optional[TimelinePlot]