From fa45ff1f8add2bbef4556f83acf20abc482c4947 Mon Sep 17 00:00:00 2001 From: Hongkuan Zhou Date: Wed, 22 Oct 2025 16:57:44 -0700 Subject: [PATCH] fix: revert change to aiperf output format (#3837) Signed-off-by: hongkuanz --- tests/planner/utils/load_generator.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/tests/planner/utils/load_generator.py b/tests/planner/utils/load_generator.py index 1ea6863ff..6ac5e38a7 100644 --- a/tests/planner/utils/load_generator.py +++ b/tests/planner/utils/load_generator.py @@ -208,16 +208,11 @@ class LoadGenerator: logger.info(f"Parsing results from: {results_file}") with open(results_file, "r") as f: - data = json.load(f) - - # aiperf now wraps metrics under "records" key - metrics = data.get("records", data) + metrics = json.load(f) results = { "throughput": metrics.get("output_token_throughput", {}).get("avg", 0), - "ttft_mean": metrics.get("ttft", {}).get( - "avg", 0 - ), # Changed from "time_to_first_token" + "ttft_mean": metrics.get("time_to_first_token", {}).get("avg", 0), "itl_mean": metrics.get("inter_token_latency", {}).get("avg", 0), "end_to_end_latency_mean": metrics.get("request_latency", {}).get( "avg", 0