pre-commit fixes

Author: wenting-zhao

commit0/cli.py

@@ -258,7 +258,12 @@ def test(
         branch = "reference"
     else:
         dataset_name = commit0_config["dataset_name"].lower()
-        if "humaneval" in dataset_name or "mbpp" in dataset_name or "bigcodebench" in dataset_name or "codecontests" in dataset_name:
+        if (
+            "humaneval" in dataset_name
+            or "mbpp" in dataset_name
+            or "bigcodebench" in dataset_name
+            or "codecontests" in dataset_name
+        ):
             branch = repo_or_repo_path
         else:
             if branch is None and not reference:

commit0/harness/build.py

@@ -28,7 +28,12 @@ def main(
     dataset_name = dataset_name.lower()
     if "swe" in dataset_name:
         dataset_type = "swebench"
-    elif "humaneval" in dataset_name or "mbpp" in dataset_name or "bigcodebench" in dataset_name or "codecontests" in dataset_name:
+    elif (
+        "humaneval" in dataset_name
+        or "mbpp" in dataset_name
+        or "bigcodebench" in dataset_name
+        or "codecontests" in dataset_name
+    ):
         dataset_type = "simple"
     else:
         dataset_type = "commit0"

commit0/harness/run_pytest_ids.py

@@ -62,7 +62,12 @@ def main(
         if "swe" in dataset_name:
             repo_name = example["instance_id"]
             dataset_type = "swebench"
-        elif "humaneval" in dataset_name or "mbpp" in dataset_name or "bigcodebench" in dataset_name or "codecontests" in dataset_name:
+        elif (
+            "humaneval" in dataset_name
+            or "mbpp" in dataset_name
+            or "bigcodebench" in dataset_name
+            or "codecontests" in dataset_name
+        ):
             repo_name = example["instance_id"]
             dataset_type = "simple"
         else:

commit0/harness/setup.py

@@ -24,7 +24,12 @@ def main(
 ) -> None:
     dataset: Iterator[RepoInstance] = load_dataset(dataset_name, split=dataset_split)  # type: ignore
     dataset_name = dataset_name.lower()
-    if "humaneval" in dataset_name or "mbpp" in dataset_name or "bigcodebench" in dataset_name or "codecontests" in dataset_name:
+    if (
+        "humaneval" in dataset_name
+        or "mbpp" in dataset_name
+        or "bigcodebench" in dataset_name
+        or "codecontests" in dataset_name
+    ):
         return
     for example in dataset:
         repo_name = example["repo"].split("/")[-1]