Unverified Commit 808a7b69 authored by lkchen's avatar lkchen Committed by GitHub
Browse files

[bench] Fix benchmark/serve.py to ignore unavailable results (#22382)


Signed-off-by: default avatarLinkun <github@lkchen.net>
parent 099c0464
......@@ -665,7 +665,7 @@ def save_to_pytorch_benchmark_format(args: argparse.Namespace,
pt_records = convert_to_pytorch_benchmark_format(
args=args,
metrics={k: [results[k]]
for k in metrics},
for k in metrics if k in results},
extra_info={
k: results[k]
for k in results if k not in metrics and k not in ignored_metrics
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment