when branch is not specified, defaults to active branch.

Author: wenting-zhao

commit0/cli.py

@@ -10,6 +10,7 @@
 import commit0.harness.lint
 import commit0.harness.save
 from commit0.harness.constants import SPLIT, SPLIT_ALL
+from commit0.harness.utils import get_active_branch
 import subprocess
 import yaml
 import os
@@ -245,14 +246,11 @@ def test(
 
     commit0_config = read_commit0_dot_file(commit0_dot_file_path)
 
-    if not branch and not reference:
-        raise typer.BadParameter(
-            f"Invalid {highlight('BRANCH', Colors.RED)}. Either --reference or provide a branch name.",
-            param_hint="BRANCH",
-        )
     if reference:
         branch = "reference"
-    assert branch is not None, "branch is not specified"
+    if branch is None and not reference:
+        git_path = os.path.join(commit0_config["base_dir"], repo_or_repo_path.split("/")[-1])
+        branch = get_active_branch(git_path)
 
     if verbose == 2:
         typer.echo(f"Running tests for repository: {repo_or_repo_path}")
@@ -294,14 +292,8 @@ def evaluate(
 ) -> None:
     """Evaluate Commit0 split you choose in Setup Stage."""
     check_commit0_path()
-    if not branch and not reference:
-        raise typer.BadParameter(
-            f"Invalid {highlight('BRANCH', Colors.RED)}. Either --reference or provide a branch name",
-            param_hint="BRANCH",
-        )
     if reference:
         branch = "reference"
-    assert branch is not None, "branch is not specified"
 
     commit0_config = read_commit0_dot_file(commit0_dot_file_path)
     check_valid(commit0_config["repo_split"], SPLIT)

commit0/harness/evaluate.py

@@ -10,7 +10,7 @@
 from commit0.harness.run_pytest_ids import main as run_tests
 from commit0.harness.get_pytest_ids import main as get_tests
 from commit0.harness.constants import RepoInstance, SPLIT, RUN_PYTEST_LOG_DIR
-from commit0.harness.utils import get_hash_string
+from commit0.harness.utils import get_hash_string, get_active_branch
 
 logging.basicConfig(
     level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
@@ -40,6 +40,9 @@ def main(
             continue
         pairs.append((repo_name, example["test"]["test_dir"]))
         hashed_test_ids = get_hash_string(example["test"]["test_dir"])
+        if branch is None:
+            git_path = os.path.join(base_dir, repo_name)
+            branch = get_active_branch(git_path)
         log_dir = RUN_PYTEST_LOG_DIR / repo_name / branch / hashed_test_ids
         log_dirs.append(str(log_dir))
 

commit0/harness/utils.py

@@ -6,7 +6,7 @@
 import time
 import sys
 from pathlib import Path
-from typing import Optional
+from typing import Optional, Union
 
 from fastcore.net import HTTP404NotFoundError, HTTP403ForbiddenError  # type: ignore
 from ghapi.core import GhApi
@@ -189,4 +189,28 @@ def generate_patch_between_commits(
         raise Exception(f"Error generating patch: {e}")
 
 
+def get_active_branch(repo_path: Union[str, Path]) -> str:
+    """
+    Retrieve the current active branch of a Git repository.
+
+    Args:
+        repo_path (Path): The path to git repo.
+
+    Returns:
+        str: The name of the active branch.
+
+    Raises:
+        Exception: If the repository is in a detached HEAD state.
+    """
+    repo = git.Repo(repo_path)
+    try:
+        # Get the current active branch
+        branch = repo.active_branch.name
+    except TypeError as e:
+        raise Exception(f"{e}\nThis means the repository is in a detached HEAD state. "
+                        "To proceed, please specify a valid branch.")
+
+    return branch
+
+
 __all__ = []