Skip to content

Commit fb1ebcb

Browse files
committed
refactor to better handle creating profile generator
1 parent 9a0c582 commit fb1ebcb

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

src/guidellm/executor/executor.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,13 +14,16 @@ def __init__(
1414
self,
1515
request_generator: RequestGenerator,
1616
backend: Backend,
17-
profile_mode: Union[str, ProfileGenerationModes] = "fixed_rate",
17+
rate_type: str = "sweep",
1818
profile_args: Optional[Dict[str, Any]] = None,
1919
max_requests: Optional[int] = None,
2020
max_duration: Optional[float] = None,
2121
):
2222
self.request_generator = request_generator
2323
self.backend = backend
24+
profile_mode = "sweep"
25+
if rate_type in {"synchronous", "constant", "poisson"}:
26+
profile_mode = "fixed_rate"
2427
self.profile = ProfileGenerator.create_generator(
2528
profile_mode, **(profile_args or {})
2629
)

src/guidellm/main.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,11 +107,10 @@ def main(
107107
raise ValueError(f"Unknown data type: {data_type}")
108108

109109
# Create executor
110-
profile_mode = rate_type if rate_type == "sweep" else "fixed_rate"
111110
executor = Executor(
112111
request_generator=request_generator,
113112
backend=backend,
114-
profile_mode=profile_mode,
113+
rate_type=rate_type,
115114
profile_args={"rate_type": rate_type, "rate": rate},
116115
max_requests=num_requests,
117116
max_duration=num_seconds,

0 commit comments

Comments
 (0)