From a55fb2b6b1ea4ad71f646906428a57fbc2386f4d Mon Sep 17 00:00:00 2001
From: Robert Shaw <ubuntu@ip-10-10-40-78.ec2.internal>
Date: Wed, 8 May 2024 19:11:22 +0000
Subject: [PATCH] updated test

---
 .github/scripts/run-tests | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.github/scripts/run-tests b/.github/scripts/run-tests
index 046059e4088..e383145a40d 100755
--- a/.github/scripts/run-tests
+++ b/.github/scripts/run-tests
@@ -117,6 +117,8 @@ do
         CUDA_VISIBLE_DEVICES=0,1 pytest ${CC_PYTEST_FLAGS} --junitxml=${RESULT_XML} ${TEST} || LOCAL_SUCCESS=$?
     elif [[ "${TEST}" == *"test_models_logprobs"* ]]; then
         pytest --forked ${CC_PYTEST_FLAGS} --junitxml=${RESULT_XML} ${TEST} || LOCAL_SUCCESS=$?
+    elif [[ "${TEST}" == *"basic_correctness/test_preemption"* ]]; then
+        VLLM_TEST_ENABLE_ARTIFICIAL_PREEMPT=1 pytest ${CC_PYTEST_FLAGS} --junitxml=${RESULT_XML} ${TEST} || LOCAL_SUCCESS=$?
     else
         pytest ${CC_PYTEST_FLAGS} --junitxml=${RESULT_XML} ${TEST} || LOCAL_SUCCESS=$?
     fi