From e9fdb75b90223339137eadb33f854faa030845a4 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Tue, 23 May 2023 18:10:55 +0200
Subject: [PATCH] do not log any artifacts if mlflow logging is disabled
---
dan/manager/training.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/dan/manager/training.py b/dan/manager/training.py
index 5821d2e8..0bd2d67a 100644
--- a/dan/manager/training.py
+++ b/dan/manager/training.py
@@ -871,8 +871,9 @@ class GenericTrainingManager:
with open(path, "w") as f:
yaml.dump(metrics, stream=f)
- # Log mlflow artifacts
- mlflow.log_artifact(path, "predictions")
+ if mlflow_logging:
+ # Log mlflow artifacts
+ mlflow.log_artifact(path, "predictions")
def output_pred(self, name):
path = os.path.join(
--
GitLab