Merge branch 'develop'

Author: floriankrb

pyproject.toml

@@ -41,7 +41,7 @@ classifiers = [
 
 dependencies = [
     "anemoi-utils[provenance]>=0.1.7",
-    "zarr",
+    "zarr<=2.17.0",
     "pyyaml",
     "numpy",
     "tqdm",

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

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()