saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix bug in print test stats

Browse source 
Apparently GIT_DEFAULT_BRACH can be empty, it looks like `github.event.repository.default_branch` is not populated for schedule-triggered workflows
Pull Request resolved: https://github.com/pytorch/pytorch/pull/74349
Approved by: https://github.com/seemethere, https://github.com/janeyx99
This commit is contained in:
Michael Suo 2022-03-16 23:01:53 +00:00 committed by PyTorch MergeBot
parent b604314aa6
commit e1d2d3f480
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tools/stats/print_test_stats.py
View file

@ -107,9 +107,14 @@ def plural(n: int) -> str:
def get_base_commit(sha1: str) -> str:
default_branch = f"origin/{os.environ.get('GIT_DEFAULT_BRANCH', 'master')}"
default_branch = os.environ.get('GIT_DEFAULT_BRANCH')
# capture None and "" cases
if not default_branch:
default_branch = "master"
default_remote = f"origin/{default_branch}"
return subprocess.check_output(
["git", "merge-base", sha1, default_branch],
["git", "merge-base", sha1, default_remote],
encoding="ascii",
).strip()