Skip to content
Snippets Groups Projects
Commit d5cfdda5 authored by Yoann Schneider's avatar Yoann Schneider :tennis:
Browse files

Do not log any artifacts if mlflow logging is disabled

parent 9168b38e
No related branches found
No related tags found
1 merge request!134Do not log any artifacts if mlflow logging is disabled
......@@ -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(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment