fixes

Author: wenting-zhao

agent/run_agent.py

@@ -134,7 +134,7 @@ def run_agent_for_repo(
         local_repo, "HEAD", example["base_commit"]
     )
     # Call the commit0 get-tests command to retrieve test files
-    test_files_str = get_tests(repo_name, verbose=0)
+    test_files_str = [xx for x in get_tests(repo_name, verbose=0) for xx in x]
     test_files = sorted(list(set([i.split(":")[0] for i in test_files_str])))
 
     # prepare the log dir

agent/run_agent_no_rich.py

@@ -96,7 +96,7 @@ def run_agent_for_repo(
         local_repo, "HEAD", example["base_commit"]
     )
     # Call the commit0 get-tests command to retrieve test files
-    test_files_str = get_tests(repo_name, verbose=0)
+    test_files_str = [xx for x in get_tests(repo_name, verbose=0) for xx in x]
     test_files = sorted(list(set([i.split(":")[0] for i in test_files_str])))
 
     # prepare the log dir

commit0/harness/evaluate.py

@@ -34,7 +34,7 @@ def main(
     dataset: Iterator[RepoInstance] = load_dataset(dataset_name, split=dataset_split)  # type: ignore
     if "swe" in dataset_name.lower():
         if repo_split == "all":
-            repos = dataset["instance_id"]
+            repos = dataset["instance_id"]  # type: ignore
         else:
             repos = [repo_split]
     else:

commit0/harness/get_pytest_ids.py

@@ -1,19 +1,20 @@
 import bz2
-from typing import List, Tuple
+from typing import List
 import commit0
 import os
 
 
-def read(bz2_file):
+def read(bz2_file: str) -> str:
     with bz2.open(bz2_file, "rt") as f:
         out = f.read()
     return out
 
-def main(repo: str, verbose: int) -> Tuple[List[str], List[str]]:
+
+def main(repo: str, verbose: int) -> List[List[str]]:
     repo = repo.lower()
     repo = repo.replace(".", "-")
     commit0_path = os.path.dirname(commit0.__file__)
-    if '__' in repo:
+    if "__" in repo:
         in_file_fail = read(f"{commit0_path}/data/test_ids/{repo}#fail_to_pass.bz2")
         in_file_pass = read(f"{commit0_path}/data/test_ids/{repo}#pass_to_pass.bz2")
     else:

commit0/harness/run_pytest_ids.py

@@ -116,7 +116,7 @@ def main(
         local_repo, example["base_commit"], commit_id
     )
     patch_file = Path(log_dir / "patch.diff")
-    patch_file.write_text(patch, encoding='utf-8', errors='ignore')
+    patch_file.write_text(patch, encoding="utf-8", errors="ignore")
 
     # make eval file
     if coverage: