From 7720ecd8201931e271ea78f047e7809fa0096ff2 Mon Sep 17 00:00:00 2001 From: Derek Whatley Date: Wed, 13 Nov 2024 16:22:07 -0500 Subject: [PATCH] Resolve key error in retrieval profiling data parser --- .../profile_data_parser/image_retrieval_profile_data_parser.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/genai-perf/genai_perf/profile_data_parser/image_retrieval_profile_data_parser.py b/genai-perf/genai_perf/profile_data_parser/image_retrieval_profile_data_parser.py index 1184ac8e..561cb1c9 100755 --- a/genai-perf/genai_perf/profile_data_parser/image_retrieval_profile_data_parser.py +++ b/genai-perf/genai_perf/profile_data_parser/image_retrieval_profile_data_parser.py @@ -66,7 +66,8 @@ def _parse_requests(self, requests: dict) -> ImageRetrievalMetrics: req_latency_ns = res_timestamps[-1] - req_timestamp request_latencies.append(req_latency_ns) - contents = load_json_str(req_inputs["input"]) + payload = load_json_str(req_inputs["payload"]) + contents = payload["input"] num_images = len([c for c in contents if c["type"] == "image_url"]) # image throughput