Feature/prepml (#56)

* fix setting mars type
ecmwf · Nov 26, 2024 · 33d750f · 33d750f
1 parent 1496c6a
commit 33d750f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/anemoi/inference/grib/encoding.py b/src/anemoi/inference/grib/encoding.py
@@ -39,7 +39,6 @@ def grib_keys(
     date,
     time,
     step,
-    type,
     keys,
     quiet,
     grib1_keys={},
@@ -86,7 +85,11 @@ def grib_keys(
         if edition == 2:
             result.update(grib2_keys.get(param, {}))
 
+    result.setdefault("type", "fc")
+    type = result.get("type")
+
     if type is not None:
+        # For organisations that do not use type
         result.setdefault("dataType", type)
 
     # if stream is not None:
@@ -160,7 +163,7 @@ def same(w, v, k):
             w = handle.get(k)
 
         if not same(w, v, k):
-            mismatches[k] = (w, v)
+            mismatches[k] = 'Expected "{}" but got "{}"'.format(v, w)
 
     if mismatches:
         raise ValueError(f"GRIB field could not be encoded. Mismatches={mismatches}")
diff --git a/src/anemoi/inference/outputs/grib.py b/src/anemoi/inference/outputs/grib.py
@@ -70,7 +70,6 @@ def write_initial_state(self, state):
                 date=None,
                 time=None,
                 step=0,
-                type="fc",
                 keys=self.encoding,
                 grib1_keys=self.grib1_keys,
                 grib2_keys=self.grib2_keys,
@@ -124,7 +123,6 @@ def write_state(self, state):
                 time=reference_date.hour,
                 step=step,
                 param=param,
-                type="fc",
                 accumulation=variable.is_accumulation,
                 keys=keys,
                 grib1_keys=self.grib1_keys,