diff --git a/dan/manager/training.py b/dan/manager/training.py
index cb0673f0e635ff2fd8c64907eafb45a90f7a9173..7e02174be28cae62d8319367269445ad9957ed3f 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(