@@ -67,7 +67,7 @@ class RunInfo(BaseModel):
6767
6868 @classmethod
6969 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" )
7171 timestamp = max (
7272 bm .run_stats .start_time for bm in benchmarks if bm .start_time is not None
7373 )
@@ -108,8 +108,8 @@ class WorkloadDetails(BaseModel):
108108
109109 @classmethod
110110 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_
113113 successful_requests = [
114114 req for bm in benchmarks for req in bm .requests .successful
115115 ]
@@ -152,13 +152,13 @@ def from_benchmarks(cls, benchmarks: list["GenerativeBenchmark"]):
152152 statistics = output_token_stats , buckets = output_token_buckets , bucket_width = 1
153153 )
154154
155- min_start_time = benchmarks [0 ].run_stats . start_time
155+ min_start_time = benchmarks [0 ].start_time
156156
157157 all_req_times = [
158- req .start_time - min_start_time
158+ req .scheduler_info . started_at - min_start_time
159159 for bm in benchmarks
160160 for req in bm .requests .successful
161- if req .start_time is not None
161+ if req .scheduler_info . started_at is not None
162162 ]
163163 number_of_buckets = len (benchmarks )
164164 request_over_time_buckets , bucket_width = Bucket .from_data (
0 commit comments