Require --quantiles set if 'quantile' in --stats. This prevents a…

… mysterious `ZeroDivisionError`. PiperOrigin-RevId: 631445591
google-research · May 7, 2024 · a050b1a · a050b1a
1 parent 1f0a1d6
commit a050b1a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scripts/compute_climatology.py b/scripts/compute_climatology.py
@@ -330,6 +330,10 @@ def _compute_seeps(kv):
     if stat not in ['seeps', 'mean']:
       for var in raw_vars:
         if stat == 'quantile':
+          if not quantiles:
+            raise ValueError(
+                'Cannot compute stat `quantile` without specifying --quantiles.'
+            )
           quantile_dim = xr.DataArray(
               quantiles, name='quantile', dims=['quantile']
           )