add ci test

Signed-off-by: zhouyu5 <[email protected]>
vllm-project · Jan 26, 2025 · 1975785 · 1975785
1 parent 6038128
commit 1975785
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.buildkite/run-hpu-test.sh b/.buildkite/run-hpu-test.sh
@@ -20,5 +20,10 @@ trap remove_docker_container_and_exit EXIT
 remove_docker_container
 
 # Run the image and launch offline inference
-docker run --runtime=habana --name=hpu-test --network=host -e HABANA_VISIBLE_DEVICES=all -e VLLM_SKIP_WARMUP=true --entrypoint="" hpu-test-env python3 examples/offline_inference/basic.py
+docker run --runtime=habana --name=hpu-test --network=host \
+    -e HABANA_VISIBLE_DEVICES=all -e VLLM_SKIP_WARMUP=true \
+    --entrypoint="" hpu-test-env \
+    /bin/bash -c "python3 examples/offline_inference/basic.py \
+    && python3 -m pip install pytest \
+    && pytest -v -s /workspace/vllm/tests/entrypoints/llm/test_guided_generate.py"
 EXITCODE=$?