From b62934531c41ee276c9e820a5b5de3e604ad1498 Mon Sep 17 00:00:00 2001 From: Jernej Sabadin Date: Mon, 25 Nov 2024 17:31:14 +0100 Subject: [PATCH] minor fixes --- luxonis_ml/tracker/tracker.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/luxonis_ml/tracker/tracker.py b/luxonis_ml/tracker/tracker.py index 6c620a74..6c773fa8 100644 --- a/luxonis_ml/tracker/tracker.py +++ b/luxonis_ml/tracker/tracker.py @@ -178,7 +178,7 @@ def store_log_locally(self, log_fn: Callable, *args, **kwargs) -> None: self.local_logs["metric"].append( {"name": args[0], "value": args[1], "step": args[2]} ) - if log_fn == self._experiment["mlflow"].log_metrics: + elif log_fn == self._experiment["mlflow"].log_metrics: self.local_logs["metrics"].append( {"metrics": args[0], "step": args[1]} ) @@ -527,7 +527,7 @@ def upload_artifact( @rank_zero_only def log_matrix(self, matrix: np.ndarray, name: str, step: int) -> None: - """Logs confusion matrix to the logging service. + """Logs matrix to the logging service. @type matrix: np.ndarray @param matrix: The confusion matrix to log.