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: ...@@ -853,8 +853,9 @@ class GenericTrainingManager:
with open(path, "w") as f: with open(path, "w") as f:
yaml.dump(metrics, stream=f) yaml.dump(metrics, stream=f)
# Log mlflow artifacts if mlflow_logging:
mlflow.log_artifact(path, "predictions") # Log mlflow artifacts
mlflow.log_artifact(path, "predictions")
def output_pred(self, name): def output_pred(self, name):
path = os.path.join( 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