pre-commit

Author: wenting-zhao

commit0/harness/evaluate.py

@@ -1,6 +1,5 @@
 import logging
 import os
-import traceback
 from collections import Counter
 
 from concurrent.futures import ThreadPoolExecutor, as_completed
@@ -43,7 +42,6 @@ def main(
         log_dir = RUN_PYTEST_LOG_DIR / repo_name / branch / hashed_test_ids
         log_dirs.append(str(log_dir))
 
-
     with tqdm(total=len(repos), smoothing=0, desc="Evaluating repos") as pbar:
         with ThreadPoolExecutor(max_workers=num_workers) as executor:
             # Create a future for running each instance

commit0/harness/run_pytest_ids.py

@@ -49,13 +49,15 @@ def main(
     dataset: Iterator[RepoInstance] = load_dataset(dataset_name, split=dataset_split)  # type: ignore
     spec = None
     example = None
+    repo_name = None
     for example in dataset:
         repo_name = example["repo"].split("/")[-1]
         if repo_name in os.path.basename(repo_or_repo_dir):
             spec = make_spec(example)
             break
     assert spec is not None, "No spec available"
     assert example is not None, "No example available"
+    assert repo_name is not None, "No repo available"
 
     hashed_test_ids = get_hash_string(test_ids)
     # set up logging
@@ -66,13 +68,15 @@ def main(
 
     try:
         local_repo = git.Repo(repo_or_repo_dir)
-    except git.exc.NoSuchPathError:
+    except git.exc.NoSuchPathError:  # type: ignore
         repo_dir = os.path.join(base_dir, repo_name)
         logger.error(f"{repo_or_repo_dir} is not a git dir, trying {repo_dir} again")
         try:
             local_repo = git.Repo(repo_dir)
-        except git.exc.NoSuchPathError:
-            raise Exception(f"{repo_dir} and {repo_or_repo_dir} are not git directories.\nUsage: commit0 test {{repo_dir}} {test_ids}")
+        except git.exc.NoSuchPathError:  # type: ignore
+            raise Exception(
+                f"{repo_dir} and {repo_or_repo_dir} are not git directories.\nUsage: commit0 test {{repo_dir}} {test_ids}"
+            )
         except Exception as e:
             raise e
     if branch == "reference":
@@ -123,7 +127,9 @@ def main(
             try:
                 pytest_exit_code = int(pytest_exit_code)
             except Exception:
-                raise Exception(f"Fail to convert pytest_exit_code {pytest_exit_code} into an integer.")
+                raise Exception(
+                    f"Fail to convert pytest_exit_code {pytest_exit_code} into an integer."
+                )
         sys.exit(pytest_exit_code)
     except EvaluationError as e:
         error_msg = (