diff --git a/dan/manager/training.py b/dan/manager/training.py
index 5821d2e804f3129a263f75da0723c3c5888c97af..0bd2d67afdc87666c9bc5ead1387906ee484d025 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(