Merge branch 'develop'

ecmwf · May 13, 2024 · 9421cbd · 9421cbd
2 parents a3dda3a + 2d529e4
commit 9421cbd
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -41,7 +41,7 @@ classifiers = [
 
 dependencies = [
     "anemoi-utils[provenance]>=0.1.7",
-    "zarr",
+    "zarr<=2.17.0",
     "pyyaml",
     "numpy",
     "tqdm",

diff --git a/src/anemoi/datasets/create/__init__.py b/src/anemoi/datasets/create/__init__.py
@@ -150,10 +150,10 @@ def create(self):
         self.additions()
         self.cleanup()
 
-    def additions(self):
-        self.init_additions()
-        self.run_additions()
-        self.finalise_additions()
+    def additions(self, delta=[1, 3, 6, 12, 24]):
+        self.init_additions(delta=delta)
+        self.run_additions(delta=delta)
+        self.finalise_additions(delta=delta)
 
     def _cache_context(self):
         from .utils import cache_context

diff --git a/tests/create/test_create.py b/tests/create/test_create.py
@@ -217,7 +217,11 @@ def test_run(name):
 
     # cache=None is using the default cache
     c = Creator(output, config=config, cache=None, overwrite=True)
-    c.create()
+    c.init()
+    c.load()
+    c.finalise()
+    c.additions(delta=[1, 3, 6, 12])
+    c.cleanup()
 
     comparer = Comparer(name, output_path=output)
     comparer.compare()