Skip to content
This repository has been archived by the owner on Dec 20, 2024. It is now read-only.

Commit

Permalink
Remove excess mlflow params (#169)
Browse files Browse the repository at this point in the history
* Remove ‘metadata.dataset.variables_metadata’ from parms to log
  • Loading branch information
HCookie authored Nov 27, 2024
1 parent fa43078 commit 112d78f
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion src/anemoi/training/diagnostics/mlflow/logger.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand Down

0 comments on commit 112d78f

Please sign in to comment.