From 9351e14c4d47e65087f4339967f0c12f85e83a5f Mon Sep 17 00:00:00 2001 From: Simon Perkins Date: Thu, 14 Sep 2023 10:51:17 +0200 Subject: [PATCH] Retain ROWID coordinates during MS conversion (#286) --- daskms/apps/convert.py | 7 ------- daskms/apps/tests/test_convert.py | 1 + 2 files changed, 1 insertion(+), 7 deletions(-) diff --git a/daskms/apps/convert.py b/daskms/apps/convert.py index fc489eeb..485c8846 100644 --- a/daskms/apps/convert.py +++ b/daskms/apps/convert.py @@ -228,10 +228,6 @@ def convert_table(self, args): datasets = reader(args.input, chunks=args.chunks) - if isinstance(in_fmt, CasaFormat): - # Drop any ROWID columns - datasets = [ds.drop_vars("ROWID", errors="ignore") for ds in datasets] - if exclude_columns := args.exclude.get("MAIN", False): datasets = [ ds.drop_vars(exclude_columns, errors="ignore") for ds in datasets @@ -273,9 +269,6 @@ def convert_table(self, args): ds.drop_vars(exclude_columns, errors="ignore") for ds in datasets ] - if isinstance(in_fmt, CasaFormat): - datasets = [ds.drop_vars("ROWID", errors="ignore") for ds in datasets] - writes.append(writer(datasets, out_store)) return writes diff --git a/daskms/apps/tests/test_convert.py b/daskms/apps/tests/test_convert.py index 795ee834..933832c7 100644 --- a/daskms/apps/tests/test_convert.py +++ b/daskms/apps/tests/test_convert.py @@ -49,6 +49,7 @@ def test_convert_application(tau_ms, format, tmp_path_factory): for ds in datasets: assert "MODEL_DATA" not in ds.data_vars assert "FLAG" in ds.data_vars + assert "ROWID" in ds.coords datasets = xds_from_storage_table(f"{str(OUTPUT)}::POINTING")