()
| 464 | |
| 465 | |
| 466 | def get_git_info(): |
| 467 | try: |
| 468 | repo = git.Repo(search_parent_directories=True) |
| 469 | repo_infos = { |
| 470 | "repo_id": str(repo), |
| 471 | "repo_sha": str(repo.head.object.hexsha), |
| 472 | "repo_branch": str(repo.active_branch), |
| 473 | "hostname": str(socket.gethostname()), |
| 474 | } |
| 475 | return repo_infos |
| 476 | except TypeError: |
| 477 | return { |
| 478 | "repo_id": None, |
| 479 | "repo_sha": None, |
| 480 | "repo_branch": None, |
| 481 | "hostname": None, |
| 482 | } |
| 483 | |
| 484 | |
| 485 | ROUGE_KEYS = ["rouge1", "rouge2", "rougeL", "rougeLsum"] |