minor fixes

luxonis · Nov 25, 2024 · b629345 · b629345
1 parent 4ccf57e
commit b629345
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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.