AMAROC-681 handle empty array in cloud base calc

MeteoSwiss · Nov 9, 2023 · 5d0fcbc · 5d0fcbc
1 parent 1c8e379
commit 5d0fcbc
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/ampycloud/layer.py b/src/ampycloud/layer.py
@@ -247,9 +247,12 @@ def _calc_base_alt(vals, lookback_perc, alt_perc):
             - int(len(vals) * lookback_perc / 100):
         ]
         n_largest_elements = vals[n_largest_idxs]
-        print("NLARGEST")
-        print(n_largest_elements)
-        print(vals)
+        if len(n_largest_elements) == 0:
+            warnings.warn(
+                'Cloud base calculation got an empty array.'
+                'Maybe check lookback percentage (is set to %i)' %lookback_perc
+            )
+            return np.nan
         return np.percentile(n_largest_elements, alt_perc)
 
     mean_comp_heights = [