pre-commit

Author: wenting-zhao

baselines/run_agent.py

@@ -52,7 +52,7 @@ def run_agent_for_repo(
     repo_name = repo_name.replace(".", "-")
 
     # Call the commit0 get-tests command to retrieve test files
-    test_files_str = get_tests(repo_name, stdout=False)
+    test_files_str = get_tests(repo_name, verbose=0)
     test_files = sorted(list(set([i.split(":")[0] for i in test_files_str])))
 
     repo_path = os.path.join(commit0_config.base_dir, repo_name)

commit0/cli.py

@@ -77,7 +77,13 @@ def build(
     ),
     dataset_split: str = typer.Option("test", help="Split of the Huggingface dataset"),
     num_workers: int = typer.Option(8, help="Number of workers"),
-    verbose: int = typer.Option(1, "--verbose", "-v", help="Set this to 2 for more logging information", count=True),
+    verbose: int = typer.Option(
+        1,
+        "--verbose",
+        "-v",
+        help="Set this to 2 for more logging information",
+        count=True,
+    ),
 ) -> None:
     """Commit0 build a repository."""
     check_valid(repo_split, SPLIT)
@@ -134,7 +140,13 @@ def test(
     reference: Annotated[
         bool, typer.Option("--reference", help="Test the reference commit.")
     ] = False,
-    verbose: int = typer.Option(1, "--verbose", "-v", help="Set this to 2 for more logging information", count=True),
+    verbose: int = typer.Option(
+        1,
+        "--verbose",
+        "-v",
+        help="Set this to 2 for more logging information",
+        count=True,
+    ),
 ) -> None:
     """Run tests on a Commit0 repository."""
     if repo_or_repo_path.endswith("/"):

commit0/harness/docker_build.py

@@ -205,6 +205,7 @@ def build_repo_images(
         client (docker.DockerClient): Docker client to use for building the images
         dataset (list): List of test specs or dataset to build images for
         max_workers (int): Maximum number of workers to use for building images
+        verbose (int): Level of verbosity
 
     Return:
     ------

commit0/harness/utils.py

@@ -27,7 +27,9 @@ def __str__(self):
         )
 
 
-def setup_logger(repo: str, log_file: Path, mode: str = "w", verbose: int = 1) -> logging.Logger:
+def setup_logger(
+    repo: str, log_file: Path, mode: str = "w", verbose: int = 1
+) -> logging.Logger:
     """Used for logging the build process of images and running containers.
     It writes logs to the log file.
     """