🐛[BUG]: DLWP-Healpix coupled dataloader did not support extracting specific channels for constants array

[ivanauyeung]

Version

0.8.0

On which installation method(s) does this occur?

Source

Describe the issue

When the healpix dataset has constants array's channels which are more than chosen channels, the inputed constants array channel size will not be matched with the model required input channel size. @yairchn

Minimum reproducible example

cd examples/weather/dlwp_healpix_coupled
python train.py --config-name="config_hpx32_coupled_dlom"

Relevant log output

Traceback (most recent call last):
  File "/global/u2/i/xxx/NV-dlesm/train.py", line 193, in <module>
    train()
  File "/global/homes/i/xxx/.local/lib/python3.10/site-packages/hydra/main.py", line 94, in decorated_main
    _run_hydra(
  File "/global/homes/i/xxx/.local/lib/python3.10/site-packages/hydra/_internal/utils.py", line 394, in _run_hydra
    _run_app(
  File "/global/homes/i/xxx/.local/lib/python3.10/site-packages/hydra/_internal/utils.py", line 457, in _run_app
    run_and_report(
  File "/global/homes/i/xxx/.local/lib/python3.10/site-packages/hydra/_internal/utils.py", line 223, in run_and_report
    raise ex
  File "/global/homes/i/xxx/.local/lib/python3.10/site-packages/hydra/_internal/utils.py", line 220, in run_and_report
    return func()
  File "/global/homes/i/xxx/.local/lib/python3.10/site-packages/hydra/_internal/utils.py", line 458, in <lambda>
    lambda: hydra.run(
  File "/global/homes/i/xxx/.local/lib/python3.10/site-packages/hydra/_internal/hydra.py", line 132, in run
    _ = ret.return_value
  File "/global/homes/i/xxx/.local/lib/python3.10/site-packages/hydra/core/utils.py", line 260, in return_value
    raise self._return_value
  File "/global/homes/i/xxx/.local/lib/python3.10/site-packages/hydra/core/utils.py", line 186, in run_job
    ret.return_value = task_function(task_cfg)
  File "/global/u2/i/xxx/NV-dlesm/train.py", line 174, in train
    trainer = instantiate(
  File "/global/homes/i/xxx/.local/lib/python3.10/site-packages/hydra/_internal/instantiate/_instantiate2.py", line 226, in instantiate
    return instantiate_node(
  File "/global/homes/i/xxx/.local/lib/python3.10/site-packages/hydra/_internal/instantiate/_instantiate2.py", line 347, in instantiate_node
    return _call_target(_target_, partial, args, kwargs, full_key)
  File "/global/homes/i/xxx/.local/lib/python3.10/site-packages/hydra/_internal/instantiate/_instantiate2.py", line 97, in _call_target
    raise InstantiationException(msg) from e
hydra.errors.InstantiationException: Error in call to target 'trainer.Trainer':
RuntimeError('Given groups=1, weight of size [136, 9, 1, 1], expected input[48, 10, 128, 128] to have 9 channels, but got 10 channels instead')
full_key: trainer

Environment details