Channel selection causes data column shape error

[ktrehaeven]

Hello world

I've noticed that if I use the --chan option and specify any custom --col name other than DATA (for example --col CORRECTED_DATA1), then I get an error saying the column has the wrong shape as below:

2025-06-03 14:32:16 - shadems - INFO - loading minmax cache from ./A3562_UHF-J1331_3140-corr3-minmax-cache.json
2025-06-03 14:32:16 - shadems - INFO - axis: UV(UVW), range (20000.0, 22000.0), discretization None
2025-06-03 14:32:16 - shadems - INFO - axis: amp(CORRECTED_DATA1), corr False, range (None, None), discretization None
2025-06-03 14:32:16 - shadems - INFO - axis: amp(CORRECTED_DATA1), corr False, range (None, None), discretization 1
2025-06-03 14:32:16 - shadems - INFO - : you have asked for 1 plots employing 3 unique datums
2025-06-03 14:32:18 - shadems - INFO - : Indexing MS and building dataframes (5659763 rows, chunk size is 5000)
Traceback (most recent call last):
File "/home/trehaeven/shadems_venv/bin/shadems", line 8, in
sys.exit(driver())
File "/home/trehaeven/shadems_venv/lib/python3.8/site-packages/shade_ms/main.py", line 578, in driver
main([a for a in sys.argv[1:]])
File "/home/trehaeven/shadems_venv/lib/python3.8/site-packages/shade_ms/main.py", line 419, in main
data_plots.get_plot_data(ms, group_cols, mytaql, ms.chan_freqs,
File "/home/trehaeven/shadems_venv/lib/python3.8/site-packages/shade_ms/data_plots.py", line 197, in get_plot_data
value = axis.get_value(group, corr, extras, flag=flag, flag_row=flag_row, chanslice=chanslice)
File "/home/trehaeven/shadems_venv/lib/python3.8/site-packages/shade_ms/data_mappers.py", line 358, in get_value
raise TypeError(f"{self.name}: unexpected column shape")
TypeError: amp:CORRECTED_DATA1:False:(None, None): unexpected column shape

Maybe this is hooman error but it seems to be genuine.