Merge pull request #67 from ecmwf-lab/develop

Allow for model to set GRIB edition

Author: HCookie

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

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,