Skip to content

Commit f41cfdb

Browse files
authored
test: Allow more time for cold start on SBSA (#101)
1 parent 460bd45 commit f41cfdb

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

ci/L0_check_health_vllm/test.sh

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,11 +31,12 @@ source ../common/util.sh
3131
pip3 install pytest==8.1.1
3232
pip3 install tritonclient[grpc]
3333

34+
rm -f *.log *.report.xml
3435
RET=0
3536

3637
function setup_model_repository {
37-
local sample_model_repo_path=${1:-"../../samples/model_repository"}
38-
rm -rf models vllm_baseline_output.pkl && mkdir -p models
38+
local sample_model_repo_path="../../samples/model_repository"
39+
rm -rf models && mkdir -p models
3940
cp -r $sample_model_repo_path/vllm_model models/vllm_opt
4041
}
4142

@@ -94,8 +95,12 @@ function test_check_health {
9495
}
9596

9697
# Test health check unspecified
98+
# Cold start on SBSA device can take longer than default 120 seconds
99+
PREV_SERVER_TIMEOUT=$SERVER_TIMEOUT
100+
SERVER_TIMEOUT=240
97101
setup_model_repository
98102
test_check_health "health_check_unspecified" "test_vllm_is_healthy"
103+
SERVER_TIMEOUT=$PREV_SERVER_TIMEOUT
99104

100105
# Test health check disabled
101106
setup_model_repository

0 commit comments

Comments
 (0)