From 13d26ccd12eb9e348b7c92b1c155bf4a6c2ee575 Mon Sep 17 00:00:00 2001 From: landmanbester Date: Tue, 19 Dec 2023 11:45:12 +0200 Subject: [PATCH] allow smoothing complex type gains --- quartical/interpolation/interpolate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/quartical/interpolation/interpolate.py b/quartical/interpolation/interpolate.py index 30123696..9a70c7dd 100644 --- a/quartical/interpolation/interpolate.py +++ b/quartical/interpolation/interpolate.py @@ -93,8 +93,8 @@ def load_and_interpolate_gains(gain_xds_lod, chain, output_directory, dask_opts) merged_xds = merged_xds.chunk({**merged_xds.dims, "antenna": 1}) if term.interp_method == "1dsmooth": - if merged_xds.TYPE != "diag_complex": - raise ValueError("Smoothing only supported for diag_complex type") + if merged_xds.TYPE not in ["diag_complex", "complex"]: + raise ValueError(f"Smoothing not suported for term type {merged_xds.TYPE}") ntime, nchan, nant, ndir, ncorr = merged_xds.gains.shape if ndir > 1: raise ValueError("Smoothing only supported for direction independent gains")