Merge pull request #66 from commit-0/agent

edit branch name

Author: nanjiangwill

agent/display.py

@@ -135,7 +135,8 @@ def __init__(self, total_repos: int):
         )
         self.layout["info"]["other_info"].split_row(
             Layout(name="backend", ratio=1),
-            Layout(name="log_dir", ratio=1),
+            Layout(name="branch", ratio=2),
+            Layout(name="log_dir", ratio=2),
         )
         self.layout["info"]["agent_info"].split_row(
             Layout(name="agent_name", ratio=1),
@@ -151,6 +152,10 @@ def __init__(self, total_repos: int):
         self.layout["info"]["other_info"]["backend"].update(
             Panel(self.backend_display, title="Backend", border_style="blue")
         )
+        self.branch_display = Text("", justify="center")
+        self.layout["info"]["other_info"]["branch"].update(
+            Panel(self.branch_display, title="Branch", border_style="blue")
+        )
         self.log_dir_display = Text("", justify="center")
         self.layout["info"]["other_info"]["log_dir"].update(
             Panel(self.log_dir_display, title="Log Directory", border_style="blue")
@@ -227,9 +232,16 @@ def update_time_display(self, time_in_seconds: int) -> None:
             Panel(self.time_display, title="Time Taken", border_style="blue")
         )
 
+    def update_branch_display(self, branch: str) -> None:
+        """Update the branch display with the given branch."""
+        self.branch_display = Text(f"{branch}", justify="center")
+        self.layout["info"]["other_info"]["branch"].update(
+            Panel(self.branch_display, title="Branch", border_style="blue")
+        )
+
     def update_backend_display(self, backend: str) -> None:
         """Update the backend display with the given backend."""
-        self.backend_display = Text(f"Backend Using: {backend}", justify="center")
+        self.backend_display = Text(f"{backend}", justify="center")
         self.layout["info"]["other_info"]["backend"].update(
             Panel(self.backend_display, title="Backend", border_style="blue")
         )

agent/run_agent.py

@@ -4,7 +4,6 @@
 from datasets import load_dataset
 from git import Repo
 from agent.agent_utils import (
-    args2string,
     create_branch,
     get_message,
     get_target_edit_files,
@@ -48,7 +47,7 @@ def run_agent_for_repo(
     agent_config: AgentConfig,
     example: RepoInstance,
     update_queue: multiprocessing.Queue,
-    branch: Optional[str] = None,
+    branch: str,
     override_previous_changes: bool = False,
     backend: str = "modal",
     log_dir: str = str(RUN_AGENT_LOG_DIR.resolve()),
@@ -88,9 +87,9 @@ def run_agent_for_repo(
             f"{agent_config.agent_name} is not implemented; please add your implementations in baselines/agents.py."
         )
 
-    # if branch_name is not provided, create a new branch name based on agent_config
-    if branch is None:
-        branch = args2string(agent_config)
+    # # if branch_name is not provided, create a new branch name based on agent_config
+    # if branch is None:
+    #     branch = args2string(agent_config)
 
     create_branch(local_repo, branch, example["base_commit"])
 
@@ -231,6 +230,7 @@ def run_agent(
             agent_config.use_spec_info,
             agent_config.use_lint_info,
         )
+        display.update_branch_display(branch)
         with multiprocessing.Manager() as manager:
             update_queue = manager.Queue()
             with multiprocessing.Pool(processes=max_parallel_repos) as pool: