fix multidim rechunk error

pangeo-forge · Mar 1, 2024 · 4780101 · 4780101
1 parent 2d0c119
commit 4780101
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pangeo_forge_recipes/rechunking.py b/pangeo_forge_recipes/rechunking.py
@@ -146,7 +146,8 @@ def _invert_meshgrid(*arrays):
         selectors[n] = tuple(selectors[n])
     xi = [a[s] for a, s in zip(arrays, selectors)]
     assert all(
-        np.equal(actual, expected).all() for actual, expected in zip(arrays, np.meshgrid(*xi))
+        np.equal(actual, expected.squeeze()).all()
+        for actual, expected in zip(arrays, np.meshgrid(*xi))
     )
     return xi