check input bins validity issue python-windrose#268

HIKassem · May 28, 2024 · e0b6134 · e0b6134
1 parent 47ab1b5
commit e0b6134
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/windrose/windrose.py b/windrose/windrose.py
@@ -376,6 +376,11 @@ def _init_plot(self, direction, var, **kwargs):
         bins = np.asarray(bins)
         nbins = len(bins)
 
+        if np.isnan(bins).any():
+            raise ValueError(
+                "Input bins contain NaN values. If bins are not explicitly "\
+                "provided as 1D array, that means var array has NaN values.")
+
         # Number of sectors
         nsector = kwargs.pop("nsector", None)
         if nsector is None: