From fb014adfc1e44c48e115b8d68f1ff417f3b88c63 Mon Sep 17 00:00:00 2001
From: Nils Brueggemann <nils.brueggemann@mpimet.mpg.de>
Date: Tue, 11 Mar 2025 12:31:23 +0100
Subject: [PATCH] pyic_view: Automatic chunking w.r.t. variable dimensions.
---
scripts/pyic_view.py | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/scripts/pyic_view.py b/scripts/pyic_view.py
index b747ec0..6595e9c 100755
--- a/scripts/pyic_view.py
+++ b/scripts/pyic_view.py
@@ -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",
--
GitLab