linter

Author: vmoens

src/ghstack/clean.py

@@ -14,7 +14,9 @@
 
 
 # Regex to match ghstack branch names: gh/{username}/{number}/{kind}
-RE_GHSTACK_BRANCH = re.compile(r"^(?:refs/(?:heads|remotes/[^/]+)/)?gh/([^/]+)/([0-9]+)/(.+)$")
+RE_GHSTACK_BRANCH = re.compile(
+    r"^(?:refs/(?:heads|remotes/[^/]+)/)?gh/([^/]+)/([0-9]+)/(.+)$"
+)
 
 
 def parse_ghstack_branch(ref: str) -> Optional[Tuple[str, GhNumber, str]]:
@@ -257,10 +259,14 @@ def main(
             pr_number, is_closed = pr_info
             pr_closed_cache[pr_number] = is_closed
             if is_closed:
-                logging.info(f"PR #{pr_number} (gh/{branch_username}/{ghnum}) is closed")
+                logging.info(
+                    f"PR #{pr_number} (gh/{branch_username}/{ghnum}) is closed"
+                )
                 orphan_branches.extend(branches)
             else:
-                logging.debug(f"PR #{pr_number} (gh/{branch_username}/{ghnum}) is still open")
+                logging.debug(
+                    f"PR #{pr_number} (gh/{branch_username}/{ghnum}) is still open"
+                )
             continue
 
         # Check cache first (handles multiple ghnums mapping to same PR)
@@ -275,7 +281,9 @@ def main(
             logging.info(f"PR #{pr_number} (gh/{branch_username}/{ghnum}) is closed")
             orphan_branches.extend(branches)
         else:
-            logging.debug(f"PR #{pr_number} (gh/{branch_username}/{ghnum}) is still open")
+            logging.debug(
+                f"PR #{pr_number} (gh/{branch_username}/{ghnum}) is still open"
+            )
 
     if not orphan_branches:
         logging.info("No orphan branches found.")
@@ -317,7 +325,7 @@ def main(
     deleted_branches: List[str] = []
 
     for i in range(0, len(orphan_branches), batch_size):
-        batch = orphan_branches[i:i + batch_size]
+        batch = orphan_branches[i : i + batch_size]
         try:
             sh.git("push", remote_name, "--delete", *batch)
             deleted_branches.extend(batch)

test/clean/basic.py.test

@@ -21,7 +21,9 @@ with captured_output() as (out, err):
     deleted_dry = gh_clean(dry_run=True)
 
 # Branches should still exist after dry run
-refs_after_dry = upstream_sh.git("for-each-ref", "refs/heads/gh/", "--format=%(refname)")
+refs_after_dry = upstream_sh.git(
+    "for-each-ref", "refs/heads/gh/", "--format=%(refname)"
+)
 assert "gh/ezyang/1/" in refs_after_dry, "Branches should still exist after dry run"
 
 # Verify dry run found the right branches
@@ -38,10 +40,14 @@ with captured_output() as (out, err):
 
 # Verify the closed PR branches were deleted
 refs_after = upstream_sh.git("for-each-ref", "refs/heads/gh/", "--format=%(refname)")
-assert "gh/ezyang/1/" not in refs_after, f"Expected branches for PR 1 to be deleted, got: {refs_after}"
+assert (
+    "gh/ezyang/1/" not in refs_after
+), f"Expected branches for PR 1 to be deleted, got: {refs_after}"
 
 # Verify the open PR branches still exist
-assert "gh/ezyang/2/" in refs_after, f"Expected branches for PR 2 to still exist, got: {refs_after}"
+assert (
+    "gh/ezyang/2/" in refs_after
+), f"Expected branches for PR 2 to still exist, got: {refs_after}"
 
 # Clean again - should find nothing to clean
 with captured_output() as (out, err):