pre-commit

Author: wenting-zhao

baselines/commit0_utils.py

@@ -2,6 +2,7 @@
 import os
 import re
 import subprocess
+from dataclasses import asdict
 from pathlib import Path
 from typing import List
 
@@ -210,12 +211,24 @@ def create_branch(repo: git.Repo, branch: str, from_commit: str) -> None:
         else:
             repo.git.checkout(from_commit)
             repo.git.checkout("-b", branch)
-    except git.exc.GitCommandError as e:
+    except git.exc.GitCommandError as e:  # type: ignore
         raise RuntimeError(f"Failed to create or switch to branch '{branch}': {e}")
 
 
 def args2string(agent_config: AgentConfig) -> str:
-    from dataclasses import asdict
+    """Converts specific fields from an `AgentConfig` object into a formatted string.
+
+    Args:
+    ----
+        agent_config (AgentConfig): A dataclass object containing configuration
+        options for an agent.
+
+    Returns:
+    -------
+        str: A string representing the selected key-value pairs from the `AgentConfig`
+        object, joined by double underscores.
+
+    """
     arg_dict = asdict(agent_config)
     result_list = []
     keys_to_collect = ["model_name", "run_tests", "use_lint_info", "use_spec_info"]
@@ -227,5 +240,5 @@ def args2string(agent_config: AgentConfig) -> str:
             else:
                 value = "0"
         result_list.append(f"{key}-{value}")
-    concatenated_string = '__'.join(result_list)
+    concatenated_string = "__".join(result_list)
     return concatenated_string

baselines/run_agent.py

@@ -59,8 +59,10 @@ def run_agent_for_repo(
     repo_path = os.path.abspath(repo_path)
     try:
         local_repo = Repo(repo_path)
-    except Exception as e:
-        raise Exception(f"{repo_path} is not a git repo. Check if base_dir is correctly specified.")
+    except Exception:
+        raise Exception(
+            f"{repo_path} is not a git repo. Check if base_dir is correctly specified."
+        )
 
     target_edit_files = get_target_edit_files(repo_path)
 
@@ -79,7 +81,7 @@ def run_agent_for_repo(
     # set it back to commit 0
     # TODO: ask user for permission
     if latest_commit.hexsha != example["base_commit"]:
-        local_repo.git.reset('--hard', example["base_commit"])
+        local_repo.git.reset("--hard", example["base_commit"])
 
     with DirContext(repo_path):
         if commit0_config is None or agent_config is None:

commit0/__main__.py

@@ -69,20 +69,28 @@ def main() -> None:
         repo_or_repo_path = sys.argv[2]
         if command == "test-reference":
             if len(sys.argv) < 4:
-                raise ValueError(f"An argument is missing for commit0 test-reference.\nUsage: commit0 test-reference {{repo_dir}} {{test_ids}}")
+                raise ValueError(
+                    "An argument is missing for commit0 test-reference.\nUsage: commit0 test-reference {repo_dir} {test_ids}"
+                )
             elif len(sys.argv) > 4:
-                raise ValueError(f"Too many arguments are passed to commit0 test-reference.\nUsage: commit0 test-reference {{repo_dir}} {{test_ids}}")
+                raise ValueError(
+                    "Too many arguments are passed to commit0 test-reference.\nUsage: commit0 test-reference {repo_dir} {test_ids}"
+                )
             branch = "reference"
             test_ids = sys.argv[3]
         else:
             if len(sys.argv) < 5:
-                raise ValueError(f"An argument is missing for commit0 test.\nUsage: commit0 test {{repo_dir}} {{branch}} {{test_ids}}")
+                raise ValueError(
+                    "An argument is missing for commit0 test.\nUsage: commit0 test {repo_dir} {branch} {test_ids}"
+                )
             elif len(sys.argv) > 5:
-                raise ValueError(f"Too many arguments are passed to commit0 test.\nUsage: commit0 test {{repo_dir}} {{branch}} {{test_ids}}")
+                raise ValueError(
+                    "Too many arguments are passed to commit0 test.\nUsage: commit0 test {repo_dir} {branch} {test_ids}"
+                )
             branch = sys.argv[3]
             test_ids = sys.argv[4]
         if branch.startswith("branch="):
-            branch = branch[len("branch="):]
+            branch = branch[len("branch=") :]
         commit0.harness.run_pytest_ids.main(
             config.dataset_name,
             config.dataset_split,
@@ -98,15 +106,23 @@ def main() -> None:
     elif command == "evaluate" or command == "evaluate-reference":
         if command == "evaluate-reference":
             if len(sys.argv) < 3:
-                raise ValueError(f"An argument is missing for commit0 evaluate-reference.\nUsage: commit0 evaluate-reference {{repo_split}}")
+                raise ValueError(
+                    "An argument is missing for commit0 evaluate-reference.\nUsage: commit0 evaluate-reference {repo_split}"
+                )
             elif len(sys.argv) > 3:
-                raise ValueError(f"Too many arguments are passed to commit0 evaluate-reference.\nUsage: commit0 evaluate-reference {{repo_split}}")
+                raise ValueError(
+                    "Too many arguments are passed to commit0 evaluate-reference.\nUsage: commit0 evaluate-reference {repo_split}"
+                )
             branch = "reference"
         else:
             if len(sys.argv) < 4:
-                raise ValueError(f"An argument is missing for commit0 evaluate.\nUsage: commit0 evaluate {{repo_split}} {{branch}}")
+                raise ValueError(
+                    "An argument is missing for commit0 evaluate.\nUsage: commit0 evaluate {repo_split} {branch}"
+                )
             elif len(sys.argv) > 4:
-                raise ValueError(f"Too many arguments are passed to commit0 evaluate.\nUsage: commit0 evaluate {{repo_split}} {{branch}}")
+                raise ValueError(
+                    "Too many arguments are passed to commit0 evaluate.\nUsage: commit0 evaluate {repo_split} {branch}"
+                )
             branch = sys.argv[3]
         commit0.harness.evaluate.main(
             config.dataset_name,

commit0/harness/setup.py

@@ -26,7 +26,7 @@ def main(
             continue
         clone_url = f"https://github.com/{example['repo']}.git"
         clone_dir = os.path.abspath(os.path.join(base_dir, repo_name))
-        local_repo = clone_repo(clone_url, clone_dir, example["base_commit"], logger)
+        clone_repo(clone_url, clone_dir, example["base_commit"], logger)
 
 
 __all__ = []