fix: Inference with crps requires additional arguments in predict_step

[jakob-schloer]

Description

The problem is described in ecmwf/anemoi-core#277. The forward function of AnemoiEnsModelEncProcDec takes additional arguments that need to be passed from anemoi-inference.

[FussyDuck]

All committers have signed the CLA.

[ssmmnn11]

lgt :-)

[HCookie]

We have a CRPS runner for this, no?
https://github.com/ecmwf/anemoi-inference/blob/main/src/anemoi/inference/runners/crps.py

[jakob-schloer]

We have a CRPS runner for this, no? https://github.com/ecmwf/anemoi-inference/blob/main/src/anemoi/inference/runners/crps.py

Good to know. Do you need to change the config for that? Simply running anemoi-inference run did not work out of the box for me.

[HCookie]

Yes, you'll need to use

runner: crps

[ssmmnn11]

But the runner was a work-around to be able to run with the old branch @gmertes

[gmertes]

I'm happy with this, the idea being that every (future) extra to predict step has to be a keyword argument and not a positional.

@HCookie what do you think?