diff --git a/release/alerts/default.py b/release/alerts/default.py index b0307f55a..6f133f1ec 100644 --- a/release/alerts/default.py +++ b/release/alerts/default.py @@ -13,6 +13,11 @@ def handle_result( artifacts: Dict, last_logs: str, team: str, + commit_url: str, + session_url: str, + runtime: float, + stable: bool, + return_code: int, ) -> Optional[str]: if not status == "finished": diff --git a/release/alerts/long_running_tests.py b/release/alerts/long_running_tests.py index 1238c3206..08e052de8 100644 --- a/release/alerts/long_running_tests.py +++ b/release/alerts/long_running_tests.py @@ -13,6 +13,11 @@ def handle_result( artifacts: Dict, last_logs: str, team: str, + commit_url: str, + session_url: str, + runtime: float, + stable: bool, + return_code: int, ) -> Optional[str]: assert test_suite == "long_running_tests" diff --git a/release/alerts/rllib_tests.py b/release/alerts/rllib_tests.py index 7678437e4..af25a2e74 100644 --- a/release/alerts/rllib_tests.py +++ b/release/alerts/rllib_tests.py @@ -13,6 +13,11 @@ def handle_result( artifacts: Dict, last_logs: str, team: str, + commit_url: str, + session_url: str, + runtime: float, + stable: bool, + return_code: int, ) -> Optional[str]: assert test_suite == "rllib_tests" diff --git a/release/alerts/tune_tests.py b/release/alerts/tune_tests.py index fe360e12f..c1f08c183 100644 --- a/release/alerts/tune_tests.py +++ b/release/alerts/tune_tests.py @@ -13,6 +13,11 @@ def handle_result( artifacts: Dict, last_logs: str, team: str, + commit_url: str, + session_url: str, + runtime: float, + stable: bool, + return_code: int, ) -> Optional[str]: assert test_suite == "tune_tests" diff --git a/release/alerts/xgboost_tests.py b/release/alerts/xgboost_tests.py index b40c17d09..528a05515 100644 --- a/release/alerts/xgboost_tests.py +++ b/release/alerts/xgboost_tests.py @@ -13,6 +13,11 @@ def handle_result( artifacts: Dict, last_logs: str, team: str, + commit_url: str, + session_url: str, + runtime: float, + stable: bool, + return_code: int, ) -> Optional[str]: assert test_suite == "xgboost_tests" diff --git a/release/e2e.py b/release/e2e.py index 15ec29ed6..3f2906ca1 100644 --- a/release/e2e.py +++ b/release/e2e.py @@ -725,6 +725,11 @@ def report_result( artifacts: Dict[Any, Any], category: str, team: str, + commit_url: str, + session_url: str, + runtime: float, + stable: bool, + return_code: int, ): # session_url: str, commit_url: str, # runtime: float, stable: bool, frequency: str, return_code: int): @@ -1626,10 +1631,11 @@ def run_test_config( ) # Add these metadata here to avoid changing SQL schema. + is_stable = test_config.get("stable", True) results["_runtime"] = runtime results["_session_url"] = session_url results["_commit_url"] = commit_url - results["_stable"] = test_config.get("stable", True) + results["_stable"] = is_stable result_queue.put( State( "END", @@ -1639,6 +1645,10 @@ def run_test_config( "last_logs": logs, "results": results, "artifacts": saved_artifacts, + "runtime": runtime, + "session_url": session_url, + "commit_url": commit_url, + "stable": is_stable, }, ) ) @@ -1956,11 +1966,12 @@ def run_test_config( exit_code = ExitCode.UNKNOWN # Add these metadata here to avoid changing SQL schema. + is_stable = test_config.get("stable", True) results = {} results["_runtime"] = runtime results["_session_url"] = session_url results["_commit_url"] = commit_url - results["_stable"] = test_config.get("stable", True) + results["_stable"] = is_stable result_queue.put( State( "END", @@ -1970,6 +1981,10 @@ def run_test_config( "last_logs": logs, "results": results, "exit_code": exit_code.value, + "runtime": runtime, + "session_url": session_url, + "commit_url": commit_url, + "stable": is_stable, }, ) ) @@ -2360,6 +2375,11 @@ def run_test( artifacts=result.get("artifacts", {}), category=category, team=team, + commit_url=result.get("commit_url", ""), + session_url=result.get("session_url", ""), + runtime=result.get("runtime", -1), + stable=result.get("stable", True), + exit_code=result.get("exit_code", ExitCode.UNKNOWN), ) if not has_errored(result):