Fix: update chunking of all dimensions to auto (except dimension) (#171)

* update chunking of all dimensions to auto (except dimension) * bump ver --------- Co-authored-by: Gerald Walter Irsiegler <gerald@Irsiegler-P14s>
Open-EO · Oct 11, 2023 · c254f6e · c254f6e
1 parent 8588f3d
commit c254f6e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/openeo_processes_dask/process_implementations/ml/curve_fitting.py b/openeo_processes_dask/process_implementations/ml/curve_fitting.py
@@ -49,8 +49,11 @@ def fit_curve(
     # so we do this to generate names locally
     parameters = {f"param_{i}": v for i, v in enumerate(parameters)}
 
+    chunking = {key: "auto" for key in data.dims if key != dimension}
+    chunking[dimension] = -1
+
     # The dimension along which to fit the curves cannot be chunked!
-    rechunked_data = data.chunk({dimension: -1})
+    rechunked_data = data.chunk(chunking)
 
     def wrapper(f):
         def _wrap(*args, **kwargs):

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "openeo-processes-dask"
-version = "2023.10.1"
+version = "2023.10.2"
 description = "Python implementations of many OpenEO processes, dask-friendly by default."
 authors = ["Lukas Weidenholzer <[email protected]>", "Sean Hoyal <[email protected]>", "Valentina Hutter <[email protected]>"]
 maintainers = ["EODC Staff <[email protected]>"]