Fix precip from ERA5

ecmwf-lab · Dec 28, 2023 · cb99db1 · cb99db1
1 parent 023e653
commit cb99db1
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/ai_models_graphcast/__init__.py b/ai_models_graphcast/__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.0.4"
+__version__ = "0.0.5"
diff --git a/ai_models_graphcast/output.py b/ai_models_graphcast/output.py
@@ -57,8 +57,16 @@ def save_output_xarray(
 
             values = np.flipud(values.reshape(fs.shape))
 
-            write(
-                values,
-                template=fs,
-                step=(time + 1) * hour_steps,
-            )
+            if param == "total_precipitation_6hr":
+                write(
+                    values,
+                    template=fs,
+                    startStep=0,
+                    endStep=(time + 1) * hour_steps,
+                )
+            else:
+                write(
+                    values,
+                    template=fs,
+                    step=(time + 1) * hour_steps,
+                )