@@ -67,7 +67,7 @@ class RunInfo(BaseModel):
67
67
68
68
@classmethod
69
69
def from_benchmarks (cls , benchmarks : list ["GenerativeBenchmark" ]):
70
- model = benchmarks [0 ].worker . backend_model or "N/A"
70
+ model = benchmarks [0 ].benchmarker . backend . get ( "model" , "N/A" )
71
71
timestamp = max (
72
72
bm .run_stats .start_time for bm in benchmarks if bm .start_time is not None
73
73
)
@@ -108,8 +108,8 @@ class WorkloadDetails(BaseModel):
108
108
109
109
@classmethod
110
110
def from_benchmarks (cls , benchmarks : list ["GenerativeBenchmark" ]):
111
- target = benchmarks [0 ].worker . backend_target
112
- rate_type = benchmarks [0 ].args . profile .type_
111
+ target = benchmarks [0 ].benchmarker . backend . get ( "target" , "N/A" )
112
+ rate_type = benchmarks [0 ].scheduler . strategy .type_
113
113
successful_requests = [
114
114
req for bm in benchmarks for req in bm .requests .successful
115
115
]
@@ -152,13 +152,13 @@ def from_benchmarks(cls, benchmarks: list["GenerativeBenchmark"]):
152
152
statistics = output_token_stats , buckets = output_token_buckets , bucket_width = 1
153
153
)
154
154
155
- min_start_time = benchmarks [0 ].run_stats . start_time
155
+ min_start_time = benchmarks [0 ].start_time
156
156
157
157
all_req_times = [
158
- req .start_time - min_start_time
158
+ req .scheduler_info . started_at - min_start_time
159
159
for bm in benchmarks
160
160
for req in bm .requests .successful
161
- if req .start_time is not None
161
+ if req .scheduler_info . started_at is not None
162
162
]
163
163
number_of_buckets = len (benchmarks )
164
164
request_over_time_buckets , bucket_width = Bucket .from_data (
0 commit comments