Skip to content
Snippets Groups Projects

Do not log any artifacts if mlflow logging is disabled

Merged Yoann Schneider requested to merge do-not-log-to-mlflow-during-training-test into main
1 file
+ 3
2
Compare changes
  • Side-by-side
  • Inline
+ 3
2
@@ -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(
Loading