File tree Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Original file line number Diff line number Diff line change @@ -31,11 +31,12 @@ source ../common/util.sh
3131pip3 install pytest==8.1.1
3232pip3 install tritonclient[grpc]
3333
34+ rm -f * .log * .report.xml
3435RET=0
3536
3637function 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
97101setup_model_repository
98102test_check_health " health_check_unspecified" " test_vllm_is_healthy"
103+ SERVER_TIMEOUT=$PREV_SERVER_TIMEOUT
99104
100105# Test health check disabled
101106setup_model_repository
You can’t perform that action at this time.
0 commit comments