replaced ds.dims[dim] with ds.sizes[dim]

Deltares · Jul 9, 2024 · f8b77ee · f8b77ee
1 parent 292bbf8
commit f8b77ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xugrid/ugrid/partitioning.py b/xugrid/ugrid/partitioning.py
@@ -381,7 +381,7 @@ def merge_partitions(partitions, merge_ugrid_chunks: bool = True):
         for dim in chunks:
             # Define a single chunk for each UGRID dimension.
             if dim in ugrid_dims:
-                chunks[dim] = (merged.dims[dim],)
+                chunks[dim] = (merged.sizes[dim],)
         merged = merged.chunk(chunks)
 
     return UgridDataset(merged, merged_grids)