From f8b77ee33233a2dd70642774de7a1d4737d12798 Mon Sep 17 00:00:00 2001
From: veenstrajelmer <60435591+veenstrajelmer@users.noreply.github.com>
Date: Tue, 9 Jul 2024 11:53:23 +0200
Subject: [PATCH] replaced ds.dims[dim] with ds.sizes[dim]

---
 xugrid/ugrid/partitioning.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/xugrid/ugrid/partitioning.py b/xugrid/ugrid/partitioning.py
index 4b42e7a6c..c57ac4fbd 100644
--- 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)