Skip to content

Commit 62c4a11

Browse files
committed
update benchmark mocker requests
Signed-off-by: guangli.bao <guangli.bao@daocloud.io>
1 parent 37e0c97 commit 62c4a11

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

src/guidellm/benchmark/output.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -690,7 +690,7 @@ def _get_benchmark_extras_headers_and_values(
690690
values: list[str] = [
691691
benchmark.benchmarker.profile.model_dump_json(),
692692
json.dumps(benchmark.benchmarker.backend),
693-
json.dumps(benchmark.benchmarker.requests["attributes"]["data"]),
693+
json.dumps(benchmark.benchmarker.requests["data"]),
694694
]
695695

696696
if len(headers) != len(values):

tests/unit/mock_benchmark.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -105,9 +105,7 @@ def mock_generative_benchmark() -> GenerativeBenchmark:
105105
benchmarker=BenchmarkerDict(
106106
profile=SynchronousProfile.create("synchronous", rate=None),
107107
requests={
108-
"attributes": {
109-
"data": "prompt_tokens=256,output_tokens=128",
110-
},
108+
"data": "prompt_tokens=256,output_tokens=128",
111109
},
112110
backend={},
113111
environment={},

0 commit comments

Comments
 (0)