Skip to content

Commit e175f2b

Browse files
committed
fix type issues
Signed-off-by: dalthecow <[email protected]>
1 parent f045501 commit e175f2b

File tree

4 files changed

+1533
-1443
lines changed

4 files changed

+1533
-1443
lines changed

src/guidellm/presentation/data_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
from pydantic import BaseModel, computed_field
77

8-
from guidellm.scheduler.strategy import SchedulingStrategy
8+
from guidellm.scheduler import SchedulingStrategy
99

1010
if TYPE_CHECKING:
1111
from guidellm.benchmark import GenerativeBenchmark
@@ -239,7 +239,7 @@ def get_strategy_display_str(cls, strategy: SchedulingStrategy):
239239
def from_benchmark(cls, bm: "GenerativeBenchmark"):
240240
rps = bm.metrics.requests_per_second.successful.mean
241241
return cls(
242-
strategy_display_str=cls.get_strategy_display_str(bm.args.strategy),
242+
strategy_display_str=cls.get_strategy_display_str(bm.scheduler.strategy),
243243
requests_per_second=rps,
244244
itl=TabularDistributionSummary.from_distribution_summary(
245245
bm.metrics.inter_token_latency_ms.successful

0 commit comments

Comments
 (0)