Skip to content

Commit 300c148

Browse files
committed
test: require V0 in vllm test
Signed-off-by: Masataro Asai <guicho2.71828@gmail.com>
1 parent 74c190d commit 300c148

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

test/backends/test_vllm.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import os
12
import pydantic
23
import pytest
34
from typing_extensions import Annotated
@@ -38,6 +39,10 @@ def session(backend):
3839
yield session
3940
session.reset()
4041

42+
43+
def test_v0_api(session):
44+
assert os.environ["VLLM_USE_V1"] == "0"
45+
4146
@pytest.mark.qualitative
4247
def test_system_prompt(session):
4348
result = session.chat(

0 commit comments

Comments
 (0)