From 292fc59d61ba97b9a79b9649e693d0cac2dea15f Mon Sep 17 00:00:00 2001 From: Michael Goin Date: Thu, 1 May 2025 22:05:04 -0600 Subject: [PATCH] [CI] Actually run tests/kv_transfer/test_disagg.py in CI (#17555) Signed-off-by: mgoin --- tests/kv_transfer/test_disagg.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/kv_transfer/test_disagg.py b/tests/kv_transfer/test_disagg.py index 5b9ea6dba4..dc948a48bf 100644 --- a/tests/kv_transfer/test_disagg.py +++ b/tests/kv_transfer/test_disagg.py @@ -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}'",