Skip to content

GitLab

Unverified Commit 2255c2c7 authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

[seq2seq] get_git_info fails gracefully (#7843) 


Co-authored-by: default avatarSam Shleifer <sshleifer@gmail.com>
parent dfa4c26b
Hide whitespace changes
Inline Side-by-side
Showing with 16 additions and 8 deletions
examples/seq2seq/utils.py
View file @ 2255c2c7
...@@ -457,14 +457,22 @@ def load_json(path): ...@@ -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"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment