Avoid error when executing benchmark util outside a git folder (#1691)

Lightning-AI · Aug 22, 2024 · 2ddf893 · 2ddf893
1 parent aaed893
commit 2ddf893
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/litgpt/api.py b/litgpt/api.py
@@ -671,9 +671,12 @@ def print_table(header, data):
         print(markdown_table)
 
     import subprocess
-    g_hash = subprocess.run(['git', 'rev-parse', '--short', 'HEAD'],
-                            capture_output=True, text=True, check=True).stdout.strip()
-    print(f"Git Commit Hash: {g_hash}")
+    try:
+        g_hash = subprocess.run(['git', 'rev-parse', '--short', 'HEAD'],
+                                capture_output=True, text=True, check=True).stdout.strip()
+        print(f"Git Commit Hash: {g_hash}")
+    except subprocess.CalledProcessError:
+        print("Git Commit Hash: N/A")
     print(f"PyTorch version: {torch.__version__}")
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     print(f"Device: {device}\n")