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