diff --git a/pangeo_forge_recipes/rechunking.py b/pangeo_forge_recipes/rechunking.py index bab9fcc9..26fbf3d6 100644 --- 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