Use travis env var for branch

leonardt · Sep 20, 2018 · 7a946ac · 7a946ac
1 parent 01c5ff4
commit 7a946ac
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/scripts/bump_version.py b/scripts/bump_version.py
@@ -11,7 +11,13 @@
 with open("VERSION", "r") as f:
     version = f.read().split(".")
 
-branch = delegator.run("git symbolic-ref --short HEAD").out.rstrip()
+travis_branch = os.environ.get("TRAVIS_BRANCH", None)
+
+if travis_branch is not None:
+    branch = travis_branch
+else:
+    branch = delegator.run("git symbolic-ref --short HEAD").out.rstrip()
+
 assert branch in ["master", "dev"], branch
 
 if branch == "master":