Some vx-bench fixes

Signed-off-by: Adam Gutglick <[email protected]>

Author: AdamGS

bench-orchestrator/bench_orchestrator/cli.py

@@ -288,7 +288,7 @@ def list_runs(
         return
 
     table = Table(title="Benchmark Runs")
-    table.add_column("Run ID", style="cyan")
+    table.add_column("Run ID", style="cyan", no_wrap=True)
     table.add_column("Label", style="green")
     table.add_column("Benchmark")
     table.add_column("Engines")
@@ -298,7 +298,7 @@ def list_runs(
     for run in runs:
         status = "[yellow]partial[/yellow]" if run.partial else "[green]complete[/green]"
         table.add_row(
-            run.run_id[:30] + "..." if len(run.run_id) > 30 else run.run_id,
+            run.run_id,
             run.label or "-",
             run.benchmark,
             ", ".join(run.engines),

bench-orchestrator/bench_orchestrator/storage/schema.py

@@ -131,6 +131,7 @@ def to_dict(self) -> dict[str, Any]:
 
     @classmethod
     def from_dict(cls, data: dict[str, Any]) -> "RunMetadata":
+        print(f"partial: {data.get('partial')}")
         return cls(
             run_id=data["run_id"],
             timestamp=datetime.fromisoformat(data["timestamp"]),

bench-orchestrator/bench_orchestrator/storage/store.py

@@ -81,8 +81,8 @@ def __exit__(self, exc_type, exc_val, exc_tb) -> None:
         # Mark as partial if there was an exception
         if exc_type is not None:
             self.metadata.partial = True
-        else:
-            self.metadata.completed_at = datetime.now()
+
+        self.metadata.completed_at = datetime.now()
 
         # Write metadata
         with open(self.run_dir / "metadata.json", "w") as f:
@@ -146,7 +146,6 @@ def create_run(self, config: RunConfig, build_config: BuildConfig) -> Iterator[R
             env_triple=env_triple,
             rustflags=build_config.rustflags,
             profile=build_config.profile,
-            partial=True,  # Will be set to False on successful completion
         )
 
         ctx = RunContext(run_dir, metadata)