diff --git a/src/mednet/scripts/upload.py b/src/mednet/scripts/upload.py index f233865cc20282744591f09684f6df4086b39446..d8b1a4c2a8e583cd6f559abf1abc3c950c4686a8 100644 --- a/src/mednet/scripts/upload.py +++ b/src/mednet/scripts/upload.py @@ -181,3 +181,5 @@ def upload( mlflow.log_artifact(evaluation_file) mlflow.log_artifact(evaluation_meta_file) mlflow.log_artifact(evaluation_log_file) + # delete temporary file as no need it after logging. + train_model_temp_file.unlink()