Unverified Commit d4629dc4 authored by lkchen's avatar lkchen Committed by GitHub
Browse files

[Misc] Add __str__ for RequestStatus (#19780)


Signed-off-by: default avatarLinkun Chen <github@lkchen.net>
parent 6e9cc73f
# SPDX-License-Identifier: Apache-2.0
from vllm.v1.request import RequestStatus
def test_request_status_fmt_str():
"""Test that the string representation of RequestStatus is correct."""
assert f"{RequestStatus.WAITING}" == "WAITING"
assert f"{RequestStatus.WAITING_FOR_FSM}" == "WAITING_FOR_FSM"
assert f"{RequestStatus.WAITING_FOR_REMOTE_KVS}" == "WAITING_FOR_REMOTE_KVS"
assert f"{RequestStatus.RUNNING}" == "RUNNING"
assert f"{RequestStatus.PREEMPTED}" == "PREEMPTED"
assert f"{RequestStatus.FINISHED_STOPPED}" == "FINISHED_STOPPED"
assert f"{RequestStatus.FINISHED_LENGTH_CAPPED}" == "FINISHED_LENGTH_CAPPED"
assert f"{RequestStatus.FINISHED_ABORTED}" == "FINISHED_ABORTED"
assert f"{RequestStatus.FINISHED_IGNORED}" == "FINISHED_IGNORED"
...@@ -171,6 +171,9 @@ class RequestStatus(enum.IntEnum): ...@@ -171,6 +171,9 @@ class RequestStatus(enum.IntEnum):
FINISHED_ABORTED = enum.auto() FINISHED_ABORTED = enum.auto()
FINISHED_IGNORED = enum.auto() FINISHED_IGNORED = enum.auto()
def __str__(self):
return self.name
@staticmethod @staticmethod
def is_finished(status: "RequestStatus") -> bool: def is_finished(status: "RequestStatus") -> bool:
return status > RequestStatus.PREEMPTED return status > RequestStatus.PREEMPTED
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment