Skip to content

Commit d4629dc

Browse files
authored
[Misc] Add __str__ for RequestStatus (#19780)
Signed-off-by: Linkun Chen <github@lkchen.net>
1 parent 6e9cc73 commit d4629dc

File tree

2 files changed

+18
-0
lines changed

2 files changed

+18
-0
lines changed

tests/v1/test_request.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
# SPDX-License-Identifier: Apache-2.0
2+
from vllm.v1.request import RequestStatus
3+
4+
5+
def test_request_status_fmt_str():
6+
"""Test that the string representation of RequestStatus is correct."""
7+
assert f"{RequestStatus.WAITING}" == "WAITING"
8+
assert f"{RequestStatus.WAITING_FOR_FSM}" == "WAITING_FOR_FSM"
9+
assert f"{RequestStatus.WAITING_FOR_REMOTE_KVS}" == "WAITING_FOR_REMOTE_KVS"
10+
assert f"{RequestStatus.RUNNING}" == "RUNNING"
11+
assert f"{RequestStatus.PREEMPTED}" == "PREEMPTED"
12+
assert f"{RequestStatus.FINISHED_STOPPED}" == "FINISHED_STOPPED"
13+
assert f"{RequestStatus.FINISHED_LENGTH_CAPPED}" == "FINISHED_LENGTH_CAPPED"
14+
assert f"{RequestStatus.FINISHED_ABORTED}" == "FINISHED_ABORTED"
15+
assert f"{RequestStatus.FINISHED_IGNORED}" == "FINISHED_IGNORED"

vllm/v1/request.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,9 @@ class RequestStatus(enum.IntEnum):
171171
FINISHED_ABORTED = enum.auto()
172172
FINISHED_IGNORED = enum.auto()
173173

174+
def __str__(self):
175+
return self.name
176+
174177
@staticmethod
175178
def is_finished(status: "RequestStatus") -> bool:
176179
return status > RequestStatus.PREEMPTED

0 commit comments

Comments
 (0)