Remove excess metadata from variables (#201)

Fix remove `metadata.dataset.specific.forward.forward.attrs.variables_metadata.` from logger params
ecmwf · Dec 13, 2024 · 318c14c · 318c14c
1 parent a2d8e6d
commit 318c14c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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: