From 0fb606735c33ee7fadf4cc0b4f9c1c8ece9d7632 Mon Sep 17 00:00:00 2001 From: Gert Mertes Date: Wed, 10 Jul 2024 11:23:11 +0000 Subject: [PATCH] Comment --- src/anemoi/training/diagnostics/mlflow/auth.py | 2 +- src/anemoi/training/diagnostics/mlflow/logger.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/anemoi/training/diagnostics/mlflow/auth.py b/src/anemoi/training/diagnostics/mlflow/auth.py index 8c24adbd..c00d5772 100644 --- a/src/anemoi/training/diagnostics/mlflow/auth.py +++ b/src/anemoi/training/diagnostics/mlflow/auth.py @@ -66,7 +66,7 @@ def refresh_token(self): @refresh_token.setter def refresh_token(self, value): self._refresh_token = value - self.refresh_expires = time.time() + (self.refresh_expire_days * 86400) + self.refresh_expires = time.time() + (self.refresh_expire_days * 86400) # 86400 seconds in a day def enabled(fn): """Decorator to call or ignore a function based on the `enabled` flag.""" diff --git a/src/anemoi/training/diagnostics/mlflow/logger.py b/src/anemoi/training/diagnostics/mlflow/logger.py index 4bee1e9e..8507333b 100644 --- a/src/anemoi/training/diagnostics/mlflow/logger.py +++ b/src/anemoi/training/diagnostics/mlflow/logger.py @@ -367,10 +367,10 @@ def log_hyperparams(self, params: Union[dict[str, Any], Namespace]) -> None: @rank_zero_only def finalize(self, status: str = "success") -> None: - # finalize logging and system metrics monitor - + # save the last obtained refresh token to disk self.auth.save() + # finalize logging and system metrics monitor if run_id_to_system_metrics_monitor: run_id_to_system_metrics_monitor[self.run_id].finish() if run_id_to_log_monitor: