diff --git a/sectionate/hydro.py b/sectionate/hydro.py index 5845470..80ac0d2 100644 --- a/sectionate/hydro.py +++ b/sectionate/hydro.py @@ -97,4 +97,9 @@ def extract_1pt(da, uvpoint, xdim=xdim, ydim=ydim): # concat over new dimension hydro = xr.concat([hydro, interp_data], dim=section) + # transpose + hydro = hydro.transpose(*(..., section)) + # rechunk + hydro = hydro.chunk({section: len(hydro[section])}) + return hydro