WCS Plotter

changelog/73.trivial.rst

@@ -0,0 +1 @@
+Catch errors when animating plots with missing data and replace them with warnings.

mpl_animators/wcs.py

@@ -8,6 +8,7 @@
 
 from mpl_animators.extern import modest_image
 from .base import ArrayAnimator
+import warnings
 
 __all__ = ['ArrayAnimatorWCS']
 
@@ -277,8 +278,13 @@ def update_plot_1d(self, val, line, slider):
 
         # If we are not setting ylim globally then we set it per frame.
         if self.ylim == 'dynamic':
-            self.axes.set_ylim(float(self.data[self.frame_index].min()),
-                               float(self.data[self.frame_index].max()))
+            vmin = float(np.nanmin(self.data[self.frame_index]))
+            vmax = float(np.nanmax(self.data[self.frame_index]))
+            if np.isnan(vmin) or np.isnan(vmax):
+                warnings.warn(UserWarning(f"No data found for data slice {self.frame_index} - cannot set ylim"))
+                return
+
+            self.axes.set_ylim(vmin, vmax)
         slider.cval = val
 
     def plot_start_image_2d(self, ax):
@@ -315,7 +321,12 @@ def _get_2d_plot_limits(self):
         Get vmin, vmax of a data slice when clip_interval is specified.
         """
         percent_limits = self.clip_interval.to('%').value
-        vmin, vmax = AsymmetricPercentileInterval(*percent_limits).get_limits(self.data_transposed)
+        if np.isnan(self.data_transposed).all():
+            warnings.warn(UserWarning(f"No data found for data slice {self.frame_index} - cannot set vmin, vmax"))
+            vmin, vmax = 0, 0
+        else:
+            vmin, vmax = AsymmetricPercentileInterval(*percent_limits).get_limits(self.data_transposed)
+
         return vmin, vmax
 
     def update_plot_2d(self, val, im, slider):