open ifile with dask. supply name to stacked dataarray to avaoid obsc…

…ure default name like "stack-23412h4234kj"
dtcenter · Aug 15, 2023 · 4ae4719 · 4ae4719
1 parent 0636546
commit 4ae4719
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/metplotpy/contributed/fv3_physics_tend/cross_section_vert.py b/metplotpy/contributed/fv3_physics_tend/cross_section_vert.py
@@ -109,7 +109,7 @@ def main():
 
     # Open input file
     logging.debug("open %s", ifile)
-    fv3ds = xarray.open_dataset(ifile)
+    fv3ds = xarray.open_mfdataset(ifile)
 
     if subtract:
         logging.info("subtracting %s", subtract)
@@ -176,7 +176,7 @@ def main():
 
     # Stack variables along new tendency dimension of new DataArray.
     tendency_dim = f"{variable} tendency"
-    tendencies_avg = tendencies_avg.to_array(dim=tendency_dim)
+    tendencies_avg = tendencies_avg.to_array(dim=tendency_dim, name=tendency_dim)
     # Assign long_names to a new DataArray coordinate.
     # It will have the same shape as tendency dimension.
     tendencies_avg = tendencies_avg.assign_coords(

diff --git a/metplotpy/contributed/fv3_physics_tend/planview_fv3.py b/metplotpy/contributed/fv3_physics_tend/planview_fv3.py
@@ -117,7 +117,7 @@ def main():
 
     # Open input file
     logging.debug("open %s", ifile)
-    fv3ds = xarray.open_dataset(ifile)
+    fv3ds = xarray.open_mfdataset(ifile)
 
     if subtract:
         logging.info("subtracting %s", subtract)
@@ -184,7 +184,7 @@ def main():
 
     # Stack variables along new tendency dimension of new DataArray.
     tendency_dim = f"{variable} tendency"
-    tendencies_avg = tendencies_avg.to_array(dim=tendency_dim)
+    tendencies_avg = tendencies_avg.to_array(dim=tendency_dim, name=tendency_dim)
     # Assign long_names to a new DataArray coordinate.
     # It will have the same shape as tendency dimension.
     tendencies_avg = tendencies_avg.assign_coords(

diff --git a/metplotpy/contributed/fv3_physics_tend/vert_profile_fv3.py b/metplotpy/contributed/fv3_physics_tend/vert_profile_fv3.py
@@ -104,7 +104,7 @@ def main():
 
     # Open input file
     logging.debug("open %s", ifile)
-    fv3ds = xarray.open_dataset(ifile)
+    fv3ds = xarray.open_mfdataset(ifile)
 
     if subtract:
         logging.info("subtracting %s", subtract)
@@ -171,7 +171,7 @@ def main():
 
     # Stack variables along new tendency dimension of new DataArray.
     tendency_dim = f"{variable} tendency"
-    tendencies_avg = tendencies_avg.to_array(dim=tendency_dim)
+    tendencies_avg = tendencies_avg.to_array(dim=tendency_dim, name=tendency_dim)
     # Assign long_names to a new DataArray coordinate.
     # It will have the same shape as tendency dimension.
     tendencies_avg = tendencies_avg.assign_coords(