GRIB2 input

ecmwf-lab · Sep 23, 2024 · a51463b · a51463b
1 parent 03d1f22
commit a51463b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ai_models_graphcast/output.py b/src/ai_models_graphcast/output.py
@@ -39,9 +39,9 @@ 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", default=0)
+            param, level = fs.metadata("shortName"), fs.metadata("levelist", default=None)
 
-            if level != 0:
+            if level is not None:
                 param = GRIB_TO_XARRAY_PL.get(param, param)
                 if param not in target_variables:
                     continue