[mergebot] Add ci-no-td label on revert (#139218)

Just in case?
Pull Request resolved: https://github.com/pytorch/pytorch/pull/139218
Approved by: https://github.com/wdvr
This commit is contained in:
Catherine Lee
2024-10-30 21:36:08 +00:00
committed by PyTorch MergeBot
parent 8840889c3f
commit 24c9683355

View File

@ -1139,7 +1139,10 @@ class GitHubPR:
if label_base in label:
count += 1
full_label = f"{label_base}X{count}"
gh_add_labels(self.org, self.project, self.pr_num, [full_label], dry_run)
self.add_label(full_label, dry_run)
def add_label(self, label: str, dry_run: bool) -> None:
gh_add_labels(self.org, self.project, self.pr_num, [label], dry_run)
def merge_into(
self,
@ -1944,6 +1947,7 @@ def do_revert_prs(
)
pr.add_numbered_label("reverted", dry_run)
pr.add_label("ci-no-td", dry_run)
if not dry_run:
gh_post_commit_comment(pr.org, pr.project, commit_sha, revert_msg)
gh_update_pr_state(pr.org, pr.project, pr.pr_num)