frozenleaves/trl
1
0
Fork 0
mirror of https://github.com/huggingface/trl.git synced 2025-10-20 18:43:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix CI slow tests: ImportError: vLLM is not installed (#4287)

Browse Source
This commit is contained in:
Albert Villanova del Moral
2025-10-15 18:15:36 +02:00
committed by GitHub
parent ef40c047aa
commit c7c041ecc8
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_vllm_client_server.py
View File

@ -22,7 +22,7 @@ from transformers.testing_utils import require_torch_multi_accelerator, torch_de
from trl.extras.vllm_client import VLLMClient
from trl.scripts.vllm_serve import chunk_list
from .testing_utils import TrlTestCase, kill_process, require_3_accelerators
from .testing_utils import TrlTestCase, kill_process, require_3_accelerators, require_vllm
class TestChunkList(TrlTestCase):
@ -53,6 +53,7 @@ class TestChunkList(TrlTestCase):
@pytest.mark.slow
@require_torch_multi_accelerator
@require_vllm
class TestVLLMClientServer(TrlTestCase):
model_id = "Qwen/Qwen2.5-1.5B"
@ -212,6 +213,7 @@ class TestVLLMClientServerBaseURL(TrlTestCase):
@pytest.mark.slow
@require_3_accelerators
@require_vllm
class TestVLLMClientServerTP(TrlTestCase):
model_id = "Qwen/Qwen2.5-1.5B"
@ -274,6 +276,7 @@ class TestVLLMClientServerTP(TrlTestCase):
@pytest.mark.slow
@require_3_accelerators
@require_vllm
class TestVLLMClientServerDP(TrlTestCase):
model_id = "Qwen/Qwen2.5-1.5B"
@ -336,6 +339,7 @@ class TestVLLMClientServerDP(TrlTestCase):
@pytest.mark.slow
@require_torch_multi_accelerator
@require_vllm
class TestVLLMClientServerDeviceParameter(TrlTestCase):
"""Test the device parameter functionality in init_communicator."""