@@ -26,27 +26,27 @@ docker run --privileged --net host --shm-size=16G -it \
2626 && tpu-info \
2727 && { \
2828 echo TEST_0: Running test_perf.py; \
29- pytest -s -v /workspace/vllm/tests/tpu/test_perf.py; \
29+ python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_perf.py; \
3030 echo TEST_0_EXIT_CODE: \$ ?; \
3131 } & \
32- && { \
32+ { \
3333 echo TEST_1: Running test_compilation.py; \
34- pytest -s -v /workspace/vllm/tests/tpu/test_compilation.py; \
34+ python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_compilation.py; \
3535 echo TEST_1_EXIT_CODE: \$ ?; \
3636 } & \
3737 { \
3838 echo TEST_2: Running test_basic.py; \
39- pytest -s -v /workspace/vllm/tests/v1/tpu/test_basic.py; \
39+ python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_basic.py; \
4040 echo TEST_2_EXIT_CODE: \$ ?; \
4141 } & \
4242 { \
4343 echo TEST_3: Running test_accuracy.py::test_lm_eval_accuracy_v1_engine; \
44- pytest -s -v /workspace/vllm/tests/entrypoints/llm/test_accuracy.py::test_lm_eval_accuracy_v1_engine; \
44+ python3 -m pytest -s -v /workspace/vllm/tests/entrypoints/llm/test_accuracy.py::test_lm_eval_accuracy_v1_engine; \
4545 echo TEST_3_EXIT_CODE: \$ ?; \
4646 } & \
4747 { \
4848 echo TEST_4: Running test_quantization_accuracy.py; \
49- pytest -s -v /workspace/vllm/tests/tpu/test_quantization_accuracy.py; \
49+ python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_quantization_accuracy.py; \
5050 echo TEST_4_EXIT_CODE: \$ ?; \
5151 } & \
5252 { \
@@ -56,43 +56,43 @@ docker run --privileged --net host --shm-size=16G -it \
5656 } & \
5757 { \
5858 echo TEST_6: Running test_tpu_model_runner.py; \
59- pytest -s -v /workspace/vllm/tests/tpu/worker/test_tpu_model_runner.py; \
59+ python3 -m pytest -s -v /workspace/vllm/tests/tpu/worker/test_tpu_model_runner.py; \
6060 echo TEST_6_EXIT_CODE: \$ ?; \
6161 } & \
62- && { \
62+ { \
6363 echo TEST_7: Running test_sampler.py; \
64- pytest -s -v /workspace/vllm/tests/v1/tpu/test_sampler.py; \
64+ python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_sampler.py; \
6565 echo TEST_7_EXIT_CODE: \$ ?; \
6666 } & \
67- && { \
67+ { \
6868 echo TEST_8: Running test_topk_topp_sampler.py; \
69- pytest -s -v /workspace/vllm/tests/v1/tpu/test_topk_topp_sampler.py; \
69+ python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_topk_topp_sampler.py; \
7070 echo TEST_8_EXIT_CODE: \$ ?; \
7171 } & \
72- && { \
72+ { \
7373 echo TEST_9: Running test_multimodal.py; \
74- pytest -s -v /workspace/vllm/tests/v1/tpu/test_multimodal.py; \
74+ python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_multimodal.py; \
7575 echo TEST_9_EXIT_CODE: \$ ?; \
7676 } & \
77- && { \
77+ { \
7878 echo TEST_10: Running test_pallas.py; \
79- pytest -s -v /workspace/vllm/tests/v1/tpu/test_pallas.py; \
79+ python3 -m pytest -s -v /workspace/vllm/tests/v1/tpu/test_pallas.py; \
8080 echo TEST_10_EXIT_CODE: \$ ?; \
8181 } & \
82- && { \
82+ { \
8383 echo TEST_11: Running test_struct_output_generate.py; \
84- pytest -s -v /workspace/vllm/tests/v1/entrypoints/llm/test_struct_output_generate.py; \
84+ python3 -m pytest -s -v /workspace/vllm/tests/v1/entrypoints/llm/test_struct_output_generate.py; \
8585 echo TEST_11_EXIT_CODE: \$ ?; \
8686 } & \
87- && { \
87+ { \
8888 echo TEST_12: Running test_moe_pallas.py; \
89- pytest -s -v /workspace/vllm/tests/tpu/test_moe_pallas.py; \
89+ python3 -m pytest -s -v /workspace/vllm/tests/tpu/test_moe_pallas.py; \
9090 echo TEST_12_EXIT_CODE: \$ ?; \
9191 } & \
9292 # Disable the TPU LoRA tests until the feature is activated
93- # && { \
93+ # & { \
9494 # echo TEST_13: Running test_moe_pallas.py; \
95- # pytest -s -v /workspace/vllm/tests/tpu/lora/; \
95+ # python3 -m pytest -s -v /workspace/vllm/tests/tpu/lora/; \
9696 # echo TEST_13_EXIT_CODE: \$ ?; \
9797 # } & \
9898 wait \
0 commit comments