better handling when no git repo is present

CDCgov · Dec 16, 2024 · 673b8f6 · 673b8f6
1 parent f6cb497
commit 673b8f6
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/pipelines/forecast_state.py b/pipelines/forecast_state.py
@@ -21,9 +21,15 @@
 
 def record_git_info(model_run_dir: Path):
     metadata_file = Path(model_run_dir, "metadata.yaml")
-    repo = Repository(os.getcwd())
-    branch_name = os.environ.get("GIT_BRANCH_NAME", Path(repo.head.name).stem)
-    commit_sha = os.environ.get("GIT_COMMIT_SHA", str(repo.head.target))
+    try:
+        repo = Repository(os.getcwd())
+        branch_name = os.environ.get(
+            "GIT_BRANCH_NAME", Path(repo.head.name).stem
+        )
+        commit_sha = os.environ.get("GIT_COMMIT_SHA", str(repo.head.target))
+    except:
+        branch_name = os.environ.get("GIT_BRANCH_NAME", "unknown")
+        commit_sha = os.environ.get("GIT_COMMIT_SHA", "unknown")
 
     metadata = {
         "branch_name": branch_name,