PlotSpectrum does not work for LAM

[JPXKQX]

What happened?

PlotSpectrum callback fails when included in LAM experiments.

What are the steps to reproduce the bug?

Train a limited area model with plot: detailed so PlotSpectrum is used.

Version

develop

Platform (OS and architecture)

x86_64 GNU/Linux

Relevant log output

File ".../anemoi-training/src/anemoi/training/diagnostics/plots.py", line 255, in compute_spectra
    coeffs_field = SHExpandGLQ(field, w=zero_w[1], zero=zero_w[0])
  File ".../conda/envs/anemoi-dev/lib/python3.10/site-packages/pyshtools/backends/shtools.py", line 212, in wrapped_func
    raise SHToolsError(_shtools_status_message(returned_values[0]))
pyshtools.backends.shtools.SHToolsError: Improper dimensions of input array.

Organisation

ECMWF