Skip to content

Commit cb36c6a

Browse files
committed
More type fixes
Signed-off-by: Jared O'Connell <joconnel@redhat.com>
1 parent 5b38f40 commit cb36c6a

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

src/guidellm/presentation/data_models.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -123,19 +123,17 @@ def from_benchmarks(cls, benchmarks: list["GenerativeBenchmark"]):
123123
for i in sample_indices
124124
]
125125
sample_outputs = [
126-
successful_requests[i].output.replace("\n", " ").replace('"', "'")
127-
if successful_requests[i].output is not None
128-
else ""
126+
req.output.replace("\n", " ").replace('"', "'") if (req := successful_requests[i]).output else ""
129127
for i in sample_indices
130128
]
131129

132130
prompt_tokens = [
133-
float(req.prompt_tokens)
131+
float(req.prompt_tokens) if req.prompt_tokens is not None else -1
134132
for bm in benchmarks
135133
for req in bm.requests.successful
136134
]
137135
output_tokens = [
138-
float(req.output_tokens)
136+
float(req.output_tokens) if req.output_tokens is not None else -1
139137
for bm in benchmarks
140138
for req in bm.requests.successful
141139
]

0 commit comments

Comments
 (0)