Align ensemble number with ECMWF GRIB conventions

ecmwf-lab · Dec 10, 2024 · daacf27 · daacf27
1 parent b04f173
commit daacf27
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/ai_models_gencast/output.py b/src/ai_models_gencast/output.py
@@ -63,13 +63,15 @@ def save_output_xarray(
                         values,
                         template=fs,
                         startStep=0,
-                        number=ensemble_member,
+                        # Offset to align with GRIB numbering, e.g. 0 is control, 1+ is ensemble
+                        number=ensemble_member + 1,
                         endStep=(time + 1) * hour_steps,
                     )
                 else:
                     write(
                         values,
                         template=fs,
                         step=(time + 1) * hour_steps,
-                        number=ensemble_member,
+                        # Offset to align with GRIB numbering, e.g. 0 is control, 1+ is ensemble
+                        number=ensemble_member + 1,
                     )