Bug fix

ecmwf-lab · Sep 15, 2024 · 03d1f22 · 03d1f22
1 parent f55a164
commit 03d1f22
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/ai_models_graphcast/input.py b/src/ai_models_graphcast/input.py
@@ -37,7 +37,7 @@
 
 
 def forcing_variables_numpy(sample, forcing_variables, dates):
-    """Generate variables from climetlabs
+    """Generate variables from earthkit-data
 
     Args:
         date (datetime): Datetime of current time step in forecast

diff --git a/src/ai_models_graphcast/output.py b/src/ai_models_graphcast/output.py
@@ -39,7 +39,7 @@ def save_output_xarray(
 
     for time in range(lead_time // hour_steps):
         for fs in all_fields[: len(all_fields) // len(lagged)]:
-            param, level = fs.metadata("shortName"), fs.metadata("levelist")
+            param, level = fs.metadata("shortName"), fs.metadata("levelist", default=0)
 
             if level != 0:
                 param = GRIB_TO_XARRAY_PL.get(param, param)