File tree Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Original file line number Diff line number Diff 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 )
Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments