Merge pull request #7 from raphaeldussin/hydro_fix_output

fix dims and chunking
MOM6-community · Aug 12, 2022 · bfc1786 · bfc1786
2 parents 5f58136 + c75e240
commit bfc1786
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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