Skip to content

Commit 53250a6

Browse files
committed
server: bench: create a script to wrap all operations
1 parent ea0581b commit 53250a6

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

examples/server/bench/bench.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,8 +68,10 @@ def main(args_in: list[str] | None = None) -> None:
6868
data = json.load(bench_results)
6969
for metric_name in data['metrics']:
7070
for metric_metric in data['metrics'][metric_name]:
71-
github_env.write(
72-
f"{escape_metric_name(metric_name)}_{escape_metric_name(metric_metric)}={round(data['metrics'][metric_name][metric_metric], 2)}\n")
71+
value = data['metrics'][metric_name][metric_metric]
72+
if isinstance(value, float):
73+
github_env.write(
74+
f"{escape_metric_name(metric_name)}_{escape_metric_name(metric_metric)}={round(value, 2)}\n")
7375
token_seconds = data['metrics']['llamacpp_tokens_second']['avg']
7476
bench_results_str = json.dumps(data)
7577

0 commit comments

Comments
 (0)