fix(ecmwf-mars): Write ensemble_stat dim correctly

openclimatefix · Jan 17, 2025 · c63995d · c63995d
1 parent d301a9f
commit c63995d
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/src/nwp_consumer/internal/repositories/raw_repositories/ecmwf_mars.py b/src/nwp_consumer/internal/repositories/raw_repositories/ecmwf_mars.py
@@ -343,12 +343,11 @@ def _convert(path: pathlib.Path) -> ResultE[list[xr.DataArray]]:
             )
             del dss
 
-            if "ens-stat" in path.as_posix():
-                # Add in missing coordinates for mean/std data
-                if "enfo-es" in path.name:
-                    ds = ds.expand_dims(dim={"ensemble_stat": ["std"]})
-                elif "enfo-em" in path.name:
-                    ds = ds.expand_dims(dim={"ensemble_stat": ["mean"]})
+            # Add in missing coordinates for mean/std data
+            if "enfo-es" in path.name:
+                ds = ds.expand_dims(dim={"ensemble_stat": ["std"]})
+            elif "enfo-em" in path.name:
+                ds = ds.expand_dims(dim={"ensemble_stat": ["mean"]})
             da: xr.DataArray = (
                 ds
                 .pipe(