Skip to content
Snippets Groups Projects
Commit fb014adf authored by Nils Brüggemann's avatar Nils Brüggemann
Browse files

pyic_view: Automatic chunking w.r.t. variable dimensions.

parent 6bb05cb4
No related branches found
Tags 0.4.0
No related merge requests found
Pipeline #99520 passed
......@@ -115,8 +115,15 @@ def get_data(ds, var_name, it, iz, res, lon_reg, lat_reg):
depth_name = 'none'
#else:
# raise ValueError(f"::: Unknown number of dimensions for {var_name}: {ds[var_name].shape}")
da = ds[var_name]
chunks = dict()
if 'time' in da.dims:
chunks = dict(time=1)
if depth_name!='none':
chunks[depth_name] = 1
da = da.chunk(**chunks)
dai = pyic.interp_to_rectgrid_xr(
ds[var_name].isel(**isel_dict), res=res,
da.isel(**isel_dict), res=res,
lon_reg=lon_reg, lat_reg=lat_reg,
verbose=False,
mask_out_of_range=False,
......@@ -408,7 +415,6 @@ class view(object):
self.message(self.flist)
self.ds = xr.open_mfdataset(
self.flist, **mfdset_kwargs,
chunks=dict(time=1, depth=1, depth_2=1)
)
delvars = [
"clon_bnds", "clat_bnds", "elon_bnds", "elat_bnds",
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment