From d929d4bb210574fb63861c3e6489c3bf117a3548 Mon Sep 17 00:00:00 2001
From: Dong Zhou <Zhou.Dong@microsoft.com>
Date: Mon, 1 Nov 2021 09:29:44 +0000
Subject: [PATCH] rm recorder temp file

---
 qlib/workflow/recorder.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/qlib/workflow/recorder.py b/qlib/workflow/recorder.py
index 0bf6f4841..73da5e19f 100644
--- a/qlib/workflow/recorder.py
+++ b/qlib/workflow/recorder.py
@@ -1,6 +1,7 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
+import os
 from qlib.utils.serial import Serializable
 import mlflow, logging
 import shutil, os, pickle, tempfile, codecs, pickle
@@ -333,7 +334,9 @@ class MLflowRecorder(Recorder):
         try:
             path = self.client.download_artifacts(self.id, name)
             with Path(path).open("rb") as f:
-                return pickle.load(f)
+                data = pickle.load(f)
+            os.remove(path)
+            return data
         except Exception as e:
             raise LoadObjectError(message=str(e))