From 808a7b69df479b6b3a16181711cac7ca28a9b941 Mon Sep 17 00:00:00 2001 From: lkchen Date: Thu, 7 Aug 2025 23:15:50 -0700 Subject: [PATCH] [bench] Fix benchmark/serve.py to ignore unavailable results (#22382) Signed-off-by: Linkun --- vllm/benchmarks/serve.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/benchmarks/serve.py b/vllm/benchmarks/serve.py index 6d52b51a9fcd0..7cdf87cb4c3bc 100644 --- a/vllm/benchmarks/serve.py +++ b/vllm/benchmarks/serve.py @@ -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