Use dict(batch.sizes) instead of batch.sizes.mapping

To fix `error: "Mapping[Hashable, int]" has no attribute "mapping" [attr-defined]`, due to incorrect type hint in upstream xarray.
xarray-contrib · Jan 1, 2024 · 5dbd411 · 5dbd411
1 parent f50c5f4
commit 5dbd411
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xbatcher/testing.py b/xbatcher/testing.py
@@ -183,7 +183,7 @@ def validate_batch_dimensions(
 
     # Check the names and lengths of the dimensions are equal
     TestCase().assertDictEqual(
-        expected_dims, batch.sizes.mapping, msg="Dimension names and/or lengths differ"
+        expected_dims, dict(batch.sizes), msg="Dimension names and/or lengths differ"
     )
     # Check the dimension order is equal
     for var in batch.data_vars: