0.2.0 - Feature release

What's Changed

This release brings some changes to the default config. Use the anemoi-training config CLI to re-generate your default config.

Added

• Add anemoi-transform link to documentation
• Codeowners file (#56)
• Changelog merge strategy (#56)

Miscellaneous

• Introduction of remapper to anemoi-models leads to changes in the data indices. Some preprocessors cannot be applied in-place anymore.

Functionality

• Enable the callback for plotting a histogram for variables containing NaNs
• Enforce same binning for histograms comparing true data to predicted data
• Fix: Inference checkpoints are now saved according the frequency settings defined in the config #37
• Feature: Add configurable models #50
• Feature: Authentication support for mlflow sync - #51
• Feature: Support training for datasets with missing time steps #48
• Feature: AnemoiMlflowClient, an mlflow client with authentication support #86
• Long Rollout Plots

Fixed

• Fix TypeError raised when trying to JSON serialise datetime.timedelta object - #43
• Bugfixes for CI (#56)
• Fix mlflow subcommand on python 3.9 #62
• Show correct subcommand in MLFlow - Addresses #39 in #61
• Fix interactive multi-GPU training #82
• Allow 500 characters in mlflow logging #88

Changed

• Updated configuration examples in documentation and corrected links - #46
• Remove credential prompt from mlflow login, replace with seed refresh token via web - #78
• Update CODEOWNERS

Commit Log

• fix: histogram only on non-nan values by @sahahner in #15
• fix: saving frequency bug for inference checkpoints by @anaprietonem in #37
• [pre-commit.ci] pre-commit autoupdate by @pre-commit-ci in #36
• feature: make models instantiateable by @JesperDramsch in #50
• Feat: adding in ability to configure precip like plots by @da-ewanp in #49
• Changes to data indices in anemoi models by @sahahner in #17
• fix: remapped variables in tests need to be dictionaries by @sahahner in #52
• Fix - datetime.timedelta to string conversion for JSON by @gareth-j in #43
• 40 support dataset with missing timesteps by @JPXKQX in #48
• Expanded intro docs and examples by @gareth-j in #46
• Chore/multiple fixes ci precommit by @theissenhelen in #56
• 23 plot a validation sample after long rollout by @sahahner in #26
• [fix] Capture Anemoi Training subcommands in MLFlow by @JesperDramsch in #61
• fix version pinning by @MeraX in #66
• Fix mlflow command on python 3.9 by @gmertes in #62
• Authentication support for mlflow sync by @gmertes in #51
• New mlflow authentication API by @gmertes in #78
• add link to transform by @b8raoult in #81
• Allow for longer truncation when mlflow > 1.28 by @HCookie in #88
• Update CODEOWNERS by @b8raoult in #90
• Fix interactive multi-GPU training by @gmertes in #82
• Add AnemoiMlflowClient with auth support by @gmertes in #86
• [pre-commit.ci] pre-commit autoupdate by @pre-commit-ci in #74
• Make pin_memory of the Dataloader configurable by @MeraX in #64

New Contributors

• @anaprietonem made their first contribution in #37
• @da-ewanp made their first contribution in #49
• @MeraX made their first contribution in #66
• @b8raoult made their first contribution in #81
• @HCookie made their first contribution in #88

Full Changelog: https://github.com/ecmwf/anemoi-training/blob/develop/CHANGELOG.md#020---feature-release---2024-10-16