add unify_chunks() to merge_partitions()

Deltares · Aug 14, 2024 · 57b4b7b · 57b4b7b
1 parent 42f10ab
commit 57b4b7b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/xugrid/ugrid/partitioning.py b/xugrid/ugrid/partitioning.py
@@ -376,6 +376,9 @@ def merge_partitions(partitions, merge_ugrid_chunks: bool = True):
             )
             merged.update(merged_selection)
 
+    # prevent inconsistent chunks, for which merged.chunks will give a ValueError
+    merged = merged.unify_chunks()
+
     # Merge chunks along the UGRID dimensions.
     if merged.chunks and merge_ugrid_chunks:
         chunks = dict(merged.chunks)