Convert formatting to use ruff instead of yapf + isort (#26247)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
@@ -6,16 +6,18 @@ from vllm.v1.metrics.stats import IterationStats
|
||||
def test_iteration_stats_repr():
|
||||
iteration_stats = IterationStats()
|
||||
iteration_stats.iteration_timestamp = 0
|
||||
expected_repr = ("IterationStats("
|
||||
"iteration_timestamp=0, "
|
||||
"num_generation_tokens=0, "
|
||||
"num_prompt_tokens=0, "
|
||||
"num_preempted_reqs=0, "
|
||||
"finished_requests=[], "
|
||||
"max_num_generation_tokens_iter=[], "
|
||||
"n_params_iter=[], "
|
||||
"time_to_first_tokens_iter=[], "
|
||||
"inter_token_latencies_iter=[], "
|
||||
"waiting_lora_adapters={}, "
|
||||
"running_lora_adapters={})")
|
||||
expected_repr = (
|
||||
"IterationStats("
|
||||
"iteration_timestamp=0, "
|
||||
"num_generation_tokens=0, "
|
||||
"num_prompt_tokens=0, "
|
||||
"num_preempted_reqs=0, "
|
||||
"finished_requests=[], "
|
||||
"max_num_generation_tokens_iter=[], "
|
||||
"n_params_iter=[], "
|
||||
"time_to_first_tokens_iter=[], "
|
||||
"inter_token_latencies_iter=[], "
|
||||
"waiting_lora_adapters={}, "
|
||||
"running_lora_adapters={})"
|
||||
)
|
||||
assert repr(iteration_stats) == expected_repr
|
||||
|
||||
Reference in New Issue
Block a user