[CI] Actually run tests/kv_transfer/test_disagg.py in CI (#17555)
Signed-off-by: mgoin <mgoin64@gmail.com>
This commit is contained in:
@ -14,8 +14,8 @@ import torch
|
||||
# Fixture to set up environment variables and teardown servers after tests
|
||||
@pytest.fixture(scope="module", autouse=True)
|
||||
def setup_servers():
|
||||
if torch.cuda.device_count() < 4:
|
||||
pytest.skip("Skipping test: fewer than 4 GPUs available")
|
||||
if torch.cuda.device_count() < 2:
|
||||
pytest.skip("Skipping test: fewer than 2 GPUs available")
|
||||
|
||||
# Set up environment variables
|
||||
VLLM_HOST_IP = subprocess.check_output("hostname -I | awk '{print $1}'",
|
||||
|
||||
Reference in New Issue
Block a user