diff --git a/src/anemoi/training/diagnostics/mlflow/logger.py b/src/anemoi/training/diagnostics/mlflow/logger.py index 71d5c475..78a80be9 100644 --- a/src/anemoi/training/diagnostics/mlflow/logger.py +++ b/src/anemoi/training/diagnostics/mlflow/logger.py @@ -501,7 +501,16 @@ def _clean_params(params: dict[str, Any]) -> dict[str, Any]: dict[str, Any] Cleaned up params ready for MlFlow. """ - prefixes_to_remove = ["hardware", "data", "dataloader", "model", "training", "diagnostics", "metadata.config"] + prefixes_to_remove = [ + "hardware", + "data", + "dataloader", + "model", + "training", + "diagnostics", + "metadata.config", + "metadata.dataset.variables_metadata", + ] keys_to_remove = [key for key in params if any(key.startswith(prefix) for prefix in prefixes_to_remove)] for key in keys_to_remove: del params[key]