From a23960c8ee7e2a6957aaa765bbeb45b0a8dfa32b Mon Sep 17 00:00:00 2001
From: Gokhan Ozbulak <gokhan.ozbulak@idiap.ch>
Date: Tue, 23 Apr 2024 13:23:53 +0200
Subject: [PATCH] Delete temp model file after it's done #60.
---
src/mednet/scripts/upload.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/src/mednet/scripts/upload.py b/src/mednet/scripts/upload.py
index f233865c..d8b1a4c2 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()
--
GitLab