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

[seq2seq] get_git_info fails gracefully (#7843)

Browse Source 
Co-authored-by: Sam Shleifer <sshleifer@gmail.com>
This commit is contained in:
Stas Bekman
2020-10-15 21:22:43 -07:00
committed by GitHub
parent dfa4c26bc0
commit 2255c2c7a0
1 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
examples/seq2seq/utils.py
View File

@@ -457,14 +457,22 @@ def load_json(path):
def get_git_info(): def get_git_info():
repo = git.Repo(search_parent_directories=True) try:
repo_infos = { repo = git.Repo(search_parent_directories=True)
"repo_id": str(repo), repo_infos = {
"repo_sha": str(repo.head.object.hexsha), "repo_id": str(repo),
"repo_branch": str(repo.active_branch), "repo_sha": str(repo.head.object.hexsha),
"hostname": str(socket.gethostname()), "repo_branch": str(repo.active_branch),
} "hostname": str(socket.gethostname()),
return repo_infos }
return repo_infos
except TypeError:
return {
"repo_id": None,
"repo_sha": None,
"repo_branch": None,
"hostname": None,
}
ROUGE_KEYS = ["rouge1", "rouge2", "rougeL", "rougeLsum"] ROUGE_KEYS = ["rouge1", "rouge2", "rougeL", "rougeLsum"]