From d5cfdda5c42ba1758ba478f13f0db3435525a860 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Wed, 24 May 2023 14:12:26 +0000
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 cb0673f0..7e02174b 100644
--- a/dan/manager/training.py
+++ b/dan/manager/training.py
@@ -853,8 +853,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