[mix-messy-graph] Remove origin after setup

Author: woojiahao

exercise_utils/git.py

@@ -73,3 +73,8 @@ def push(remote: str, branch: str, verbose: bool) -> None:
 def track_remote_branch(remote: str, branch: str, verbose: bool) -> None:
     """Tracks a remote branch locally using the same name."""
     run_command(["git", "branch", branch, f"{remote}/{branch}"], verbose)
+
+
+def remove_remote(remote: str, verbose: bool) -> None:
+    """Removes a given remote."""
+    run_command(["git", "remote", "rm", remote], verbose)

mix_messy_graph/download.py

@@ -1,10 +1,15 @@
-from exercise_utils.git import merge, merge_with_message, track_remote_branch
+from exercise_utils.git import (
+    merge_with_message,
+    remove_remote,
+    track_remote_branch,
+)
 
 
 def setup(verbose: bool = False):
     remote_name = "origin"
     remote_branches = ["feature-search", "feature-delete", "list"]
     for remote_branch_name in remote_branches:
         track_remote_branch(remote_name, remote_branch_name, verbose)
+    remove_remote("origin", verbose)
 
     merge_with_message("feature-search", False, "Merge search feature", verbose)