From 995bf1c459e844cae837c4ce9e493fe4119aaf41 Mon Sep 17 00:00:00 2001 From: Harrison Cook Date: Fri, 13 Dec 2024 11:52:36 +0000 Subject: [PATCH] Remove excess metadata from variables `metadata.dataset.specific.forward.forward.attrs.variables_metadata.` adds > 1000 params. --- src/anemoi/training/diagnostics/mlflow/logger.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/anemoi/training/diagnostics/mlflow/logger.py b/src/anemoi/training/diagnostics/mlflow/logger.py index 7b8e5f53..03a4b6de 100644 --- a/src/anemoi/training/diagnostics/mlflow/logger.py +++ b/src/anemoi/training/diagnostics/mlflow/logger.py @@ -510,6 +510,7 @@ def _clean_params(params: dict[str, Any]) -> dict[str, Any]: "diagnostics", "metadata.config", "metadata.dataset.variables_metadata", + "metadata.dataset.specific.forward.forward.attrs.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: