SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Use the PR URL in CI report (#17269)

Browse Source 
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2022-05-16 22:02:28 +02:00
committed by GitHub
parent 3fb82f74fd
commit 8600d770d4
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
utils/notification_service.py
View File

@@ -732,6 +732,9 @@ if __name__ == "__main__":
artifact_path["gpu"] artifact_path["gpu"]
] += f"*{line}*\n_{stacktraces.pop(0)}_\n\n" ] += f"*{line}*\n_{stacktraces.pop(0)}_\n\n"
# To find the PR number in a commit title, for example, `Add AwesomeFormer model (#99999)`
pr_number_re = re.compile(r"\(#(\d+)\)$")
title = f"🤗 Results of the {ci_event} tests." title = f"🤗 Results of the {ci_event} tests."
# Add PR title with a link for push CI # Add PR title with a link for push CI
ci_title = os.environ.get("CI_TITLE") ci_title = os.environ.get("CI_TITLE")
@@ -739,6 +742,13 @@ if __name__ == "__main__":
if ci_title is not None: if ci_title is not None:
assert commit_url is not None assert commit_url is not None
ci_title = ci_title.strip().split("\n")[0].strip() ci_title = ci_title.strip().split("\n")[0].strip()
# Find the PR number (if any) and change the url to the actual PR page.
numbers = pr_number_re.findall(ci_title)
if len(numbers) > 0:
pr_number = numbers[0]
commit_url = f"https://github.com/huggingface/transformers/pull/{pr_number}"
ci_title = f"<{commit_url}|{ci_title}>" ci_title = f"<{commit_url}|{ci_title}>"
else: else:
ci_title = "" ci_title = ""