diff --git a/utils/notification_service.py b/utils/notification_service.py index 9fac01c637..32df44ef3e 100644 --- a/utils/notification_service.py +++ b/utils/notification_service.py @@ -727,7 +727,7 @@ def retrieve_available_artifacts(): _available_artifacts[artifact_name].add_path(directory, gpu="single") elif artifact_name.startswith("multi-gpu"): - artifact_name = directory[len("multi-gpu") + 1 :] + artifact_name = artifact_name[len("multi-gpu") + 1 :] if artifact_name in _available_artifacts: _available_artifacts[artifact_name].multi_gpu = True @@ -985,12 +985,13 @@ if __name__ == "__main__": continue for artifact_path in available_artifacts[additional_files[key]].paths: + # Link to the GitHub Action job + job_name = key if artifact_path["gpu"] is not None: - additional_results[key]["job_link"][artifact_path["gpu"]] = github_actions_job_links.get( - f"{key} ({artifact_path['gpu']}-gpu)" - ) - else: - additional_results[key]["job_link"][artifact_path["gpu"]] = github_actions_job_links.get(key) + job_name = f"{key} ({artifact_path['gpu']}-gpu)" + if job_name_prefix: + job_name = f"{job_name_prefix} / {job_name}" + additional_results[key]["job_link"][artifact_path["gpu"]] = github_actions_job_links.get(job_name) artifact = retrieve_artifact(artifact_path["path"], artifact_path["gpu"]) stacktraces = handle_stacktraces(artifact["failures_line"])