I’ve noticed an interesting discrepancy when setting up the AIFS Ensemble v2 for inference with anemoi. Specifically, the documentation around AIFS-ENS v2 shows that w on pressure levels is a output/“diagnostic” field, specifically on its huggingface page linked below as well as in pre-launch presentations.
However, when attempting to provide inputs to the model after loading the checkpoint, an error is thrown if w is not provided on all 14 pressure levels. While I defacto must presume that the model checkpoint file itself is correct, it would be helpful to clarify that this is the case and that several of the diagnostic/output fields are actually required inputs. If that’s not the case, having a model checkpoint file with corrected metadata would instead be a great solution.
Thanks for any and all insight here.