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

8
examples/seq2seq/utils.py
View File

@@ -457,6 +457,7 @@ def load_json(path):
def get_git_info(): def get_git_info():
try:
repo = git.Repo(search_parent_directories=True) repo = git.Repo(search_parent_directories=True)
repo_infos = { repo_infos = {
"repo_id": str(repo), "repo_id": str(repo),
@@ -465,6 +466,13 @@ def get_git_info():
"hostname": str(socket.gethostname()), "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"]