aggregate: unchunk spatial dimension before apply_ufunc

PyPSA · Oct 19, 2023 · f0fdf1c · f0fdf1c
1 parent e1ad5db
commit f0fdf1c
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/.gitignore b/.gitignore
@@ -15,6 +15,7 @@ atlite.egg-info/
 doc/.vscode/settings.json
 .vscode/settings.json
 test/*.nc
+dev-scripts/
 examples/*.nc
 examples/*.csv
 examples/*.zip

diff --git a/RELEASE_NOTES.rst b/RELEASE_NOTES.rst
@@ -9,12 +9,14 @@ Release Notes
 
 
 
-.. Upcoming Release
-.. ================
+Upcoming Release
+================
 
-.. * Fix: the wind turbine power curve is checked for a missing cut-out wind speed and an option to add a
-..   cut-out wind speed at the end of the power curve is introduced. From the next release v0.2.13, adding
-..   a cut-out wind speed will be the default behavior (`GH #316 <https://github.com/PyPSA/atlite/pull/316>`_)
+
+* Fix: the wind turbine power curve is checked for a missing cut-out wind speed and an option to add a
+  cut-out wind speed at the end of the power curve is introduced. From the next release v0.2.13, adding
+  a cut-out wind speed will be the default behavior (`GH #316 <https://github.com/PyPSA/atlite/pull/316>`_)
+* Compatibility with xarray >= 2023.09.: The chunked spatial dimension in `aggregate` was raising an error with the new xarray version. This is fixed now.
 
 
 Version 0.2.11

diff --git a/atlite/aggregate.py b/atlite/aggregate.py
@@ -16,6 +16,7 @@ def aggregate_matrix(da, matrix, index):
         index = index.rename("dim_0")
     if isinstance(da.data, dask.array.core.Array):
         da = da.stack(spatial=("y", "x"))
+        da = da.chunk(dict(spatial=-1))
         return xr.apply_ufunc(
             lambda da: da * matrix.T,
             da,