mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
a37b4fa03a
with https://github.com/pytorch/test-infra/pull/3882 Add -ic/--ignore-current flag for merge. It ignores the currently failing checks but will stop when there is a new failure. If there are no pending checks, it fails and tells you to use -f/--force. Doesn't work on ghstacks with more than 1 PR. Pull Request resolved: https://github.com/pytorch/pytorch/pull/96756 Approved by: https://github.com/huydhn
89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
import os
|
|
import re
|
|
from typing import List, Pattern, Optional, Tuple
|
|
|
|
|
|
BOT_COMMANDS_WIKI = "https://github.com/pytorch/pytorch/wiki/Bot-commands"
|
|
|
|
CIFLOW_LABEL = re.compile(r"^ciflow/.+")
|
|
CIFLOW_TRUNK_LABEL = re.compile(r"^ciflow/trunk")
|
|
|
|
OFFICE_HOURS_LINK = "https://github.com/pytorch/pytorch/wiki/Dev-Infra-Office-Hours"
|
|
CONTACT_US = f"Questions? Feedback? Please reach out to the [PyTorch DevX Team]({OFFICE_HOURS_LINK})"
|
|
ALTERNATIVES = (
|
|
f"Learn more about merging in the [wiki]({BOT_COMMANDS_WIKI})."
|
|
)
|
|
|
|
|
|
def has_label(labels: List[str], pattern: Pattern[str] = CIFLOW_LABEL) -> bool:
|
|
return len(list(filter(pattern.match, labels))) > 0
|
|
|
|
|
|
class TryMergeExplainer(object):
|
|
force: bool
|
|
labels: List[str]
|
|
pr_num: int
|
|
org: str
|
|
project: str
|
|
ignore_current: bool
|
|
|
|
has_trunk_label: bool
|
|
has_ciflow_label: bool
|
|
|
|
def __init__(
|
|
self,
|
|
force: bool,
|
|
labels: List[str],
|
|
pr_num: int,
|
|
org: str,
|
|
project: str,
|
|
ignore_current: bool,
|
|
):
|
|
self.force = force
|
|
self.labels = labels
|
|
self.pr_num = pr_num
|
|
self.org = org
|
|
self.project = project
|
|
self.ignore_current = ignore_current
|
|
|
|
def _get_flag_msg(self, ignore_current_checks: Optional[List[Tuple[str, Optional[str]]]] = None) -> str:
|
|
if self.force:
|
|
return "Your change will be merged immediately since you used the force (-f) flag, " + \
|
|
"**bypassing any CI checks** (ETA: 1-5 minutes)."
|
|
elif self.ignore_current and ignore_current_checks is not None:
|
|
msg = f"Your change will be merged while ignoring the following {len(ignore_current_checks)} checks: "
|
|
msg += ', '.join(f"[{x[0]}]({x[1]})" for x in ignore_current_checks)
|
|
return msg
|
|
else:
|
|
return "Your change will be merged once all checks pass (ETA 0-4 Hours)."
|
|
|
|
|
|
def get_merge_message(
|
|
self,
|
|
ignore_current_checks: Optional[List[Tuple[str, Optional[str]]]] = None
|
|
) -> str:
|
|
title = "### Merge started"
|
|
main_message = self._get_flag_msg(ignore_current_checks)
|
|
|
|
advanced_debugging = "\n".join((
|
|
"<details><summary>Advanced Debugging</summary>",
|
|
"Check the merge workflow status ",
|
|
f"<a href=\"{os.getenv('GH_RUN_URL')}\">here</a>",
|
|
"</details>"
|
|
))
|
|
|
|
msg = title + "\n"
|
|
msg += main_message + "\n\n"
|
|
msg += ALTERNATIVES + "\n\n"
|
|
msg += CONTACT_US
|
|
msg += advanced_debugging
|
|
return msg
|
|
|
|
|
|
def get_revert_message(org: str, project: str, pr_num: int) -> str:
|
|
msg = (
|
|
"@pytorchbot successfully started a revert job."
|
|
+ f" Check the current status [here]({os.getenv('GH_RUN_URL')}).\n"
|
|
)
|
|
msg += CONTACT_US
|
|
return msg
|