diff --git a/modules/util/gfs_reader.py b/modules/util/gfs_reader.py index 8209431e131840e4be5a7125014ba10c05a57ea2..0550cb39dbe5e76158a517b2dc8e8e9e080736dd 100644 --- a/modules/util/gfs_reader.py +++ b/modules/util/gfs_reader.py @@ -295,7 +295,8 @@ def get_voxel(xr_dataset, fld_name, lon, lat, press, x_width=5, y_width=5, z_wid sub_fld = fld[y_c-y_h:y_c+y_h+1, x_c-x_h:x_c+x_h+1, p_c-p_h:p_c+p_h+1] sub_fld = sub_fld.expand_dims('channel') - sub_fld = sub_fld.assign_coords(channel=[fld_name], fakeDim2=lon_coords[x_c-x_h:x_c+x_h+1], fakeDim1=lat_coords[y_c-y_h:y_c+y_h+1], fakeDim0=plevs[p_c-p_h:p_c+p_h+1]) + sub_fld = sub_fld.assign_coords(channel=[fld_name], fakeDim2=lon_coords[x_c-x_h:x_c+x_h+1], + fakeDim1=lat_coords[y_c-y_h:y_c+y_h+1], fakeDim0=plevs[p_c-p_h:p_c+p_h+1]) return sub_fld