chore(ecmwf-mars): Modify chunking

openclimatefix · Jan 15, 2025 · c93423c · c93423c
1 parent 5ad56d0
commit c93423c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/nwp_consumer/internal/entities/modelmetadata.py b/src/nwp_consumer/internal/entities/modelmetadata.py
@@ -170,7 +170,7 @@ class Models:
         resolution="0.1 degrees",
         expected_coordinates=NWPDimensionCoordinateMap(
             init_time=[],
-            step=list(range(0, 48, 1)),
+            step=list(range(0, 49, 1)),
             variable=[
                 Parameter.DOWNWARD_LONGWAVE_RADIATION_FLUX_GL,
                 Parameter.DOWNWARD_SHORTWAVE_RADIATION_FLUX_GL,

diff --git a/src/nwp_consumer/internal/repositories/raw_repositories/ecmwf_mars.py b/src/nwp_consumer/internal/repositories/raw_repositories/ecmwf_mars.py
@@ -225,7 +225,9 @@ def repository() -> entities.RawRepositoryMetadata:
                 "ens-stat-india": entities.Models.ECMWF_ENS_STAT_0P1DEGREE.with_region("india"),
                 "ens-stat-uk": entities.Models.ECMWF_ENS_STAT_0P1DEGREE.with_region("uk"),
                 "ens-uk": entities.Models.ECMWF_ENS_0P1DEGREE.with_region("uk")\
-                    .with_chunk_count_overrides({"latitude": 1, "longitude": 1, "variable": 1}),
+                        .with_chunk_count_overrides(
+                            {"latitude": 2, "longitude": 2, "variable": 1, "ensemble_member": 5},
+                        ),
             },
         )