pre-commit

Author: wenting-zhao

commit0/harness/build.py

@@ -1,13 +1,10 @@
-import json
 import logging
-import traceback
 
 import docker
 from datasets import load_dataset
-from tqdm import tqdm
 from typing import Iterator
 
-from commit0.harness.constants import EVAL_BACKENDS, RepoInstance, SPLIT
+from commit0.harness.constants import RepoInstance, SPLIT
 from commit0.harness.docker_build import build_repo_images
 from commit0.harness.spec import make_spec
 
@@ -37,4 +34,5 @@ def main(
         client = docker.from_env()
         build_repo_images(client, specs, num_workers)
 
+
 __all__ = []

commit0/harness/run_pytest_ids.py

@@ -66,7 +66,9 @@ def main(
     else:
         local_branch = local_repo.branches[branch]
         commit_id = local_branch.commit.hexsha
-    patch = generate_patch_between_commits(local_repo, example["base_commit"], commit_id)
+    patch = generate_patch_between_commits(
+        local_repo, example["base_commit"], commit_id
+    )
     patch_file = Path(log_dir / "patch.diff")
     patch_file.write_text(patch)
 
@@ -84,7 +86,10 @@ def main(
             f"Evaluation must be from {', '.join(EVAL_BACKENDS)}, but {backend} is provided."
         )
 
-    files_to_copy = Files(eval_script={"src": eval_file, "dest": Path("/eval.sh")}, patch={"src": patch_file, "dest": Path("/patch.diff")})
+    files_to_copy = Files(
+        eval_script={"src": eval_file, "dest": Path("/eval.sh")},
+        patch={"src": patch_file, "dest": Path("/patch.diff")},
+    )
 
     try:
         with execution_context(spec, logger, timeout, files_to_copy) as context:

commit0/harness/utils.py

@@ -2,17 +2,12 @@
 import git.exc
 import hashlib
 import logging
-import socket
 import os
 import time
-import requests
-import subprocess
-import pwd
-from typing import Optional, Tuple
+from typing import Optional
 
 from fastcore.net import HTTP404NotFoundError, HTTP403ForbiddenError  # type: ignore
 from ghapi.core import GhApi
-from commit0.harness.constants import EVAL_BACKENDS
 
 
 class EvaluationError(Exception):
@@ -168,25 +163,31 @@ def create_repo_on_github(
             break
 
 
-def generate_patch_between_commits(repo: git.Repo, old_commit: str, new_commit: str) -> str:
-    """
-    Generate a patch string by comparing two specified commits.
+def generate_patch_between_commits(
+    repo: git.Repo, old_commit: str, new_commit: str
+) -> str:
+    """Generate a patch string by comparing two specified commits.
 
     Args:
+    ----
         repo (git.Repo): An instance of the git.Repo object representing the repository.
         old_commit (str): The hash or reference to the old commit.
         new_commit (str): The hash or reference to the new commit.
 
     Returns:
+    -------
         patch (str): A string containing the patch in the diff format between the two commits
 
     Raises:
+    ------
         git.GitCommandError: If there is an error while running git commands.
+
     """
     try:
-        patch = repo.git.diff(old_commit, new_commit, '--', '.', ':(exclude)spec.pdf')
-        return patch+'\n\n'
+        patch = repo.git.diff(old_commit, new_commit, "--", ".", ":(exclude)spec.pdf")
+        return patch + "\n\n"
     except git.GitCommandError as e:
         raise Exception(f"Error generating patch: {e}")
 
+
 __all__ = []