relax check of grib encoding

ai_models/__init__.py

@@ -5,4 +5,4 @@
 # granted to it by virtue of its status as an intergovernmental organisation
 # nor does it submit to any jurisdiction.
 
-__version__ = "0.3.1"
+__version__ = "0.3.2"

ai_models/outputs/__init__.py

@@ -83,11 +83,20 @@ def write(self, *args, **kwargs):
 
         handle, path = self.output.write(*args, **kwargs)
 
+        # Check that the GRIB keys are as expected
         for key, value in itertools.chain(
             self.output.grib_keys.items(), kwargs.items()
         ):
             if key in ("template",):
                 continue
+
+            # If "param" is a string, we what to compare it to the shortName
+            if key == "param":
+                try:
+                    float(value)
+                except ValueError:
+                    key = "shortName"
+
             assert str(handle.get(key)) == str(value), (key, handle.get(key), value)
 
         return handle, path