Added for saving git info

CHANGELOG.md

@@ -13,6 +13,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 - Added metric `FBetaVerboseMeasure` which extends `FBetaMeasure` to ensure compatibility with logging plugins and add some options.
 - Added three sample weighting techniques to `ConditionalRandomField` by supplying three new subclasses: `ConditionalRandomFieldWeightEmission`, `ConditionalRandomFieldWeightTrans`, and `ConditionalRandomFieldWeightLannoy`.
+- Added saving of github repo infos along with other metadata in model archives.
 
 ### Fixed
 

allennlp/common/meta.py

@@ -2,6 +2,8 @@
 from dataclasses import dataclass, asdict
 import json
 import logging
+import git
+import socket
 from typing import Union
 
 from allennlp.version import VERSION
@@ -28,10 +30,21 @@ def new(cls) -> "Meta":
 
     def to_file(self, path: Union[PathLike, str]) -> None:
         with open(path, "w") as meta_file:
-            json.dump(asdict(self), meta_file)
+            json.dump(asdict(self).update(get_git_info()), meta_file)
 
     @classmethod
     def from_path(cls, path: Union[PathLike, str]) -> "Meta":
         with open(path) as meta_file:
             data = json.load(meta_file)
         return cls(**data)
+
+
+def get_git_info():
+    repo = git.Repo(search_parent_directories=True)
+    repo_info = {
+        "repo_id": str(repo),
+        "repo_sha": str(repo.head.object.hexsha),
+        "repo_branch": str(repo.active_branch),
+        "hostname": str(socket.gethostname()),
+    }
+    return repo_info