diff --git a/make_release.py b/make_release.py index d6b15041ee1d66f9879fd52e007608fac49582ff..020f16e4686cfca3821a9a78653d18308331587f 100755 --- a/make_release.py +++ b/make_release.py @@ -7,9 +7,10 @@ import subprocess def main(): + default_branch = "main" branch = subprocess.check_output("git rev-parse --abbrev-ref HEAD", shell=True).decode().strip() - if branch != "main": - print("Aborting, not on 'main' branch.") + if branch != default_branch: + print(f"Aborting, not on '{default_branch}' branch.") return version_filepath = os.path.join(os.path.basename(os.path.dirname(__file__)), "__init__.py")