[CI/Build] [3/3] Reorganize entrypoints tests (#5966)

2024-06-30 12:58:49 +08:00
parent cff6a1fec1
commit 9d47f64eb6
17 changed files with 19 additions and 48 deletions
--- a/tests/entrypoints/openai/test_serving_chat.py
+++ b/tests/entrypoints/openai/test_serving_chat.py
@@ -1,15 +1,11 @@
 import asyncio
 from dataclasses import dataclass

-import pytest
-
 from vllm.entrypoints.openai.serving_chat import OpenAIServingChat

 MODEL_NAME = "openai-community/gpt2"
 CHAT_TEMPLATE = "Dummy chat template for testing {}"

-pytestmark = pytest.mark.openai
-

@dataclass
 class MockModelConfig: