From 298dbbdb8547d8c3f7a8278c804261bbd071c1a6 Mon Sep 17 00:00:00 2001
From: Benjamin McCormick <docterformer@protonmail.com>
Date: Thu, 20 Nov 2025 17:56:27 -0500
Subject: [PATCH] chore: coderabbit review

---
 tests/bench/beam.py    | 10 +++++-----
 tests/bench/harness.py | 21 +++++++++++++++++++--
 2 files changed, 24 insertions(+), 7 deletions(-)

diff --git a/tests/bench/beam.py b/tests/bench/beam.py
index 5d693b23..acc726fe 100644
--- a/tests/bench/beam.py
+++ b/tests/bench/beam.py
@@ -48,7 +48,7 @@ Optional arguments:
 --context-length: Context length subset to test (100K, 500K, 1M, 10M) (default: 100K)
 --conversation-ids: Comma-separated list of conversation IDs to test (default: all in context length)
 --anthropic-api-key: Anthropic API key for response judging (can be set in .env as LLM_ANTHROPIC_API_KEY)
---timeout: Timeout for deriver queue to empty in seconds (default: 10 minutes)
+    --timeout: Timeout for deriver queue to empty in seconds (default: 10 minutes (600s))
 --base-api-port: Base port for Honcho API instances (default: 8000)
 --pool-size: Number of Honcho instances in the pool (default: 1)
 --batch-size: Number of conversations to run concurrently in each batch (default: 1)
@@ -154,7 +154,7 @@ class BEAMRunner:
         self.pool_size: int = pool_size
         self.anthropic_api_key: str | None = anthropic_api_key
         self.timeout_seconds: int = (
-            timeout_seconds if timeout_seconds is not None else 10000
+            timeout_seconds if timeout_seconds is not None else 600
         )
         self.cleanup_workspace: bool = cleanup_workspace
         self.use_get_context: bool = use_get_context
@@ -507,7 +507,7 @@ Extract the ordered list of events or items mentioned in the response. Preserve
 
             # Now compute alignment and Kendall tau-b
             # Match extracted events to rubric events using LLM equivalence
-            alignment = await self._align_events(rubric, extracted_events)
+            alignment = self._align_events(rubric, extracted_events)
 
             # Compute Kendall tau-b
             tau: float
@@ -551,7 +551,7 @@ Extract the ordered list of events or items mentioned in the response. Preserve
                 "overall_reasoning": f"Evaluation failed due to error: {e}",
             }
 
-    async def _align_events(
+    def _align_events(
         self, expected_events: list[str], extracted_events: list[str]
     ) -> list[int]:
         """
@@ -1084,7 +1084,7 @@ async def main() -> int:
         "--timeout",
         type=int,
         default=None,
-        help="Timeout for deriver queue to empty in seconds (default: 10 minutes)",
+        help="Timeout for deriver queue to empty in seconds (default: 10 minutes (600s))",
     )
 
     parser.add_argument(
diff --git a/tests/bench/harness.py b/tests/bench/harness.py
index fc2da6cd..1a568023 100755
--- a/tests/bench/harness.py
+++ b/tests/bench/harness.py
@@ -218,6 +218,23 @@ class HonchoHarness:
         """
         print("Starting Redis cache server on port 6379...")
 
+        # Ensure clean state by removing any existing containers/volumes
+        subprocess.run(
+            [
+                "docker",
+                "compose",
+                "-f",
+                str(self.docker_compose_file),
+                "-p",
+                f"honcho_harness_{self.db_port}",
+                "down",
+                "--volumes",
+                "--remove-orphans",
+            ],
+            cwd=self.temp_dir,
+            capture_output=True,
+        )
+
         # Change to the temp directory and start the redis service
         result = subprocess.run(
             [
@@ -699,7 +716,8 @@ except Exception as e:
                 # More aggressive cleanup - remove containers, volumes, and orphaned containers
                 subprocess.run(
                     [
-                        "docker-compose",
+                        "docker",
+                        "compose",
                         "-f",
                         str(self.docker_compose_file),
                         "-p",
@@ -710,7 +728,6 @@ except Exception as e:
                     ],
                     cwd=self.temp_dir,
                     capture_output=True,
-                    text=True,
                 )
 
                 # Also try to remove any containers that might still be running