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