Tweak manage-cookie branch name

smkent · Feb 22, 2023 · 21aceda · 21aceda
1 parent a3f5e77
commit 21aceda
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cookie_python/manage/repo.py b/cookie_python/manage/repo.py
@@ -17,10 +17,12 @@
 
 
 class RepoSandbox:
-    def __init__(self, repo: str, dry_run: bool = False) -> None:
+    def __init__(
+        self, repo: str, dry_run: bool = False, branch: str = "manage-cookie"
+    ) -> None:
         self._stack = contextlib.ExitStack()
         self.repo = self.gh.find_repo(repo)
-        self.branch = "update-cookie"
+        self.branch = branch
         self.dry_run = dry_run
 
     def __enter__(self) -> RepoSandbox: