diff --git a/CHANGELOG.md b/CHANGELOG.md index f21209f9..124ccf50 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -13,6 +13,7 @@ Keep it human-readable, your future self will thank you! ### Fixed - Refactored callbacks. [#60](https://github.com/ecmwf/anemoi-training/pulls/60) - Updated docs [#115](https://github.com/ecmwf/anemoi-training/pull/115) + - Fix enabling LearningRateMonitor [#119](https://github.com/ecmwf/anemoi-training/pull/119) - Refactored rollout [#87](https://github.com/ecmwf/anemoi-training/pulls/87) - Enable longer validation rollout than training ### Added diff --git a/src/anemoi/training/diagnostics/callbacks/__init__.py b/src/anemoi/training/diagnostics/callbacks/__init__.py index 4b0921f1..3968da77 100644 --- a/src/anemoi/training/diagnostics/callbacks/__init__.py +++ b/src/anemoi/training/diagnostics/callbacks/__init__.py @@ -30,8 +30,7 @@ def nestedget(conf: DictConfig, key, default): - """ - Get a nested key from a DictConfig object + """Get a nested key from a DictConfig object E.g. >>> nestedget(config, "diagnostics.log.wandb.enabled", False) @@ -50,7 +49,7 @@ def nestedget(conf: DictConfig, key, default): ("training.swa.enabled", StochasticWeightAveraging), ( lambda config: nestedget(config, "diagnostics.log.wandb.enabled", False) - or nestedget(config, "diagnostics.log.mflow.enabled", False), + or nestedget(config, "diagnostics.log.mlflow.enabled", False), LearningRateMonitor, ), ]