log colorbars now use nanmin / nanmax to avoid nans causing error

Jammy2211 · Jan 18, 2024 · 9d6600b · 9d6600b
1 parent 6600a30
commit 9d6600b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/autoarray/plot/wrap/base/cmap.py b/autoarray/plot/wrap/base/cmap.py
@@ -50,12 +50,12 @@ def symmetric_cmap_from(self, symmetric_value=None):
 
     def vmin_from(self, array: np.ndarray):
         if self.config_dict["vmin"] is None:
-            return np.min(array)
+            return np.nanmin(array)
         return self.config_dict["vmin"]
 
     def vmax_from(self, array: np.ndarray):
         if self.config_dict["vmax"] is None:
-            return np.max(array)
+            return np.nanmax(array)
         return self.config_dict["vmax"]
 
     def norm_from(self, array: np.ndarray, use_log10: bool = False) -> object: