Skip to content

Commit b216179

Browse files
committed
fix: reduced number of decimals in eval output
1 parent be3d023 commit b216179

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

gptme/eval/main.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,9 @@
99
import subprocess
1010
import sys
1111
from collections import defaultdict
12+
from collections.abc import Generator
1213
from datetime import datetime
1314
from pathlib import Path
14-
from collections.abc import Generator
1515

1616
import click
1717
import multiprocessing_logging
@@ -55,10 +55,10 @@ def print_model_results(model_results: dict[str, list[ExecResult]]):
5555
run_tokens = len_tokens(result.run_stdout)
5656
result_total_tokens = gen_tokens + run_tokens
5757
print(
58-
f"{checkmark} {result.name}: {duration_result:.2f}s/{result_total_tokens}tok "
59-
f"(gen: {result.timings['gen']:.2f}s/{gen_tokens}tok, "
60-
f"run: {result.timings['run']:.2f}s/{run_tokens}tok, "
61-
f"eval: {result.timings['eval']:.2f}s)"
58+
f"{checkmark} {result.name}: {duration_result:.0f}s/{result_total_tokens}tok "
59+
f"(gen: {result.timings['gen']:.0f}s/{gen_tokens}tok, "
60+
f"run: {result.timings['run']:.0f}s/{run_tokens}tok, "
61+
f"eval: {result.timings['eval']:.0f}s)"
6262
)
6363
for case in cases:
6464
checkmark = "✅" if case.passed else "❌"
@@ -92,7 +92,7 @@ def print_model_results_table(model_results: dict[str, list[ExecResult]]):
9292
row.append(f"{checkmark} {reason}")
9393
else:
9494
row.append(
95-
f"{checkmark} {duration:.2f}s/{gen_tokens+run_tokens}tok"
95+
f"{checkmark} {duration:.0f}s/{gen_tokens+run_tokens}tok"
9696
)
9797
except StopIteration:
9898
row.append("❌ N/A")

0 commit comments

Comments
 (0)