Merge pull request #67 from ecmwf-lab/develop

Allow for model to set GRIB edition
ecmwf-lab · Dec 9, 2024 · 5c2ebf0 · 5c2ebf0
2 parents b1a202c + 3a7028b
commit 5c2ebf0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/ai_models/model.py b/src/ai_models/model.py
@@ -63,6 +63,7 @@ class Model:
     assets_extra_dir = None
     retrieve = {}  # Extra parameters for retrieve
     version = 1  # To be overriden in subclasses
+    grib_edition = 2  # Default GRIB edition
     grib_extra_metadata = {}  # Extra metadata for grib files
 
     param_level_ml = ([], [])  # param, level

diff --git a/src/ai_models/outputs/__init__.py b/src/ai_models/outputs/__init__.py
@@ -38,7 +38,7 @@ def __init__(self, owner, path, metadata, **kwargs):
 
     @cached_property
     def grib_keys(self):
-        edition = self.metadata.pop("edition", 2)
+        edition = self.metadata.pop("edition", self.owner.grib_edition)
 
         _grib_keys = dict(
             edition=edition,