NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-24 17:04:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

move table to display util

Browse source
This commit is contained in:
hjc-puro 2025-07-11 02:06:56 +00:00
parent 3e08c6d788
commit 68da3809e2
2 changed files with 56 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

28
atroposlib/envs/base.py
View file

@ -710,33 +710,9 @@ class BaseEnv(ABC):
# Print metrics table if verbose
if verbose:
print("\n" + "=" * 60)
print(f"Evaluation Results: {task_name}")
print("=" * 60)
header = (
f"|{'Groups':<20}|{'Version':<7}|{'Filter':<6}|{'n-shot':<6}|"
f"{'Metric':<10}|{' ':<3}|{'Value':<10}|{' ':<3}|{'Stderr':<10}|"
)
print(header)
print(
f"|{'-'*20}|{'-'*7}:{'-'*6}|{'-'*6}|{'-'*10}|{'-'*3}|{'-'*10}:{'-'*3}|{'-'*10}:|"
)
from atroposlib.utils.display import format_evaluation_table
# Main task row
for metric_name, metric_value in metrics.items():
clean_metric_name = metric_name.replace("eval/", "").replace("_", " ")
direction = (
"↑" if "correct" in metric_name or "acc" in metric_name else " "
)
row = (
f"|{task_name:<20}|{1:<7}|{'none':<6}|{'':<6}|{clean_metric_name:<10}|"
f"{direction:<3}|{metric_value:<10.4f}|{'±':<3}|{'0.0000':<10}|"
)
print(row)
print("=" * 60)
print(f"Evaluation completed in {end_time - start_time:.2f} seconds")
print("=" * 60 + "\n")
format_evaluation_table(task_name, metrics, start_time, end_time)
# Build the evaluation result structure
task_key = f"atropos|{task_name}|0"