From bde6d4f97c283878d7c240a01df597af02da9dea Mon Sep 17 00:00:00 2001 From: Onur ULUSOY Date: Wed, 29 May 2024 18:08:51 +0000 Subject: [PATCH] Sync with master --- upsonic/remote/on_prem.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/upsonic/remote/on_prem.py b/upsonic/remote/on_prem.py index 2f74c4d..00d9568 100644 --- a/upsonic/remote/on_prem.py +++ b/upsonic/remote/on_prem.py @@ -969,7 +969,7 @@ def os_name(self): else: return 'Unknown OS' - def add_run_history(self, key, version, cpu_usage_one_core, memory_usage, elapsed_time, type, params): + def add_run_history(self, key, version, cpu_usage_one_core, memory_usage, elapsed_time, type, params, exception_log): data = { "scope": key, "version": version, @@ -979,7 +979,8 @@ def add_run_history(self, key, version, cpu_usage_one_core, memory_usage, elapse "type": type, "python_version":f"{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}", "os_name": self.os_name(), - "params":json.dumps(params) + "params":json.dumps(params), + "exception_log":exception_log, } self._send_request("POST", "/dump_run", data) @@ -1048,8 +1049,11 @@ def normalize_params(*args, **kwargs): else: the_version = version the_type = "Succed" if succed else "Failed" + exception_log = None + if not succed: + exception_log = output try: - self.add_run_history(key, the_version, cpu_usage_for_one_core, memory_used, total_time, the_type, the_params) + self.add_run_history(key, the_version, cpu_usage_for_one_core, memory_used, total_time, the_type, the_params, exception_log) except: self._log(f"Error on adding run history, for server not supported. {key}") return output