replace .dims with .sizes (#212)

* replace .dims with .sizes * additional dims replacements
Deltares · Feb 9, 2024 · 0a41f61 · 0a41f61
1 parent ccc9e63
commit 0a41f61
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_ugrid_dataset.py b/tests/test_ugrid_dataset.py
@@ -1133,11 +1133,11 @@ def get_ugrid_fillvaluem999_startindex1_ds():
     )
 
     # add dummy face variable in order to have a face dimension in the uds
-    facevar = np.ones(shape=(ds2.dims[mesh2d_attrs["face_dimension"]]))
+    facevar = np.ones(shape=(ds2.sizes[mesh2d_attrs["face_dimension"]]))
     ds2["mesh2d_facevar"] = xr.DataArray(facevar, dims=(mesh2d_attrs["face_dimension"]))
 
     # add dummy nodevar to plot and trigger triangulation procedure
-    nodevar = np.ones(shape=(ds2.dims[mesh2d_attrs["node_dimension"]]))
+    nodevar = np.ones(shape=(ds2.sizes[mesh2d_attrs["node_dimension"]]))
     ds2["mesh2d_nodevar"] = xr.DataArray(nodevar, dims=(mesh2d_attrs["node_dimension"]))
     return ds2
 

diff --git a/xugrid/ugrid/conventions.py b/xugrid/ugrid/conventions.py
@@ -271,7 +271,7 @@ def _infer_dims(
                             f" with dimensions: {var_dims}"
                         )
             elif isinstance(key, int):
-                dim_size = ds.dims[dim]
+                dim_size = ds.sizes[dim]
                 if not dim_size == key:
                     raise UgridDimensionError(
                         f"Expected size {key} for dimension {dim} in variable "