fix(generic_utils): apply get_git_branch fix from coqui-tts

Same fix as coqui-ai/TTS#3561
idiap · Jun 26, 2024 · e57903f · e57903f
1 parent bfeee49
commit e57903f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/trainer/generic_utils.py b/trainer/generic_utils.py
@@ -37,14 +37,14 @@ def get_cuda():
     return use_cuda, device
 
 
-def get_git_branch():
+def get_git_branch() -> str:
     try:
         out = subprocess.check_output(["git", "branch"]).decode("utf8")
         current = next(line for line in out.split("\n") if line.startswith("*"))
         current.replace("* ", "")
     except subprocess.CalledProcessError:
         current = "inside_docker"
-    except FileNotFoundError:
+    except (FileNotFoundError, StopIteration):
         current = "unknown"
     return current