added extra : pip install anemoi-dataset[zarr2]

Author: floriankrb

pyproject.toml

@@ -52,15 +52,14 @@ dependencies = [
   "cfunits",
   "glom",
   "jsonschema",
-  "numcodecs<0.16",           # Until we move to zarr3
   "numpy",
   "pytest>=8.4.1",
   "pytest-xdist>=3.7",
   "pyyaml",
   "ruamel-yaml",
   "semantic-version",
   "tqdm",
-  "zarr",                     # will use zarr3 for python >=3.11
+  "zarr",
 ]
 
 optional-dependencies.all = [
@@ -118,6 +117,11 @@ optional-dependencies.xarray = [
   "s3fs",
 ]
 
+optional-dependencies.zarr2 = [
+  "numcodecs<0.16",
+  "zarr<3",
+]
+
 urls.Changelog = "https://github.com/ecmwf/anemoi-datasets/CHANGELOG.md"
 
 urls.Documentation = "https://anemoi-datasets.readthedocs.io/"