Unverified Commit f2089d2e authored by Kris Hung's avatar Kris Hung Committed by GitHub
Browse files

fix: Update pytest markers for deepep test. (#2561)

parent 6a358f7c
...@@ -47,8 +47,8 @@ while [[ $# -gt 0 ]]; do ...@@ -47,8 +47,8 @@ while [[ $# -gt 0 ]]; do
echo " --gpus-per-node L Number of GPUs per node (required, int)" echo " --gpus-per-node L Number of GPUs per node (required, int)"
echo " --master-addr ADDR Master node address (default: localhost)" echo " --master-addr ADDR Master node address (default: localhost)"
echo " --log-dir DIR Directory for log files (default: ./logs)" echo " --log-dir DIR Directory for log files (default: ./logs)"
echo " --model MODEL Model name to use (default: deepseek-ai/DeepSeek-R1)" echo " --model MODEL Model name to use (default: ${MODEL})"
echo " -h, --help Show this help message" echo " -h, --help Show this help message"
exit 0 exit 0
;; ;;
*) *)
......
...@@ -264,7 +264,11 @@ vllm_configs = { ...@@ -264,7 +264,11 @@ vllm_configs = {
name="deepep", name="deepep",
directory="/workspace/components/backends/vllm", directory="/workspace/components/backends/vllm",
script_name="dsr1_dep.sh", script_name="dsr1_dep.sh",
marks=[pytest.mark.gpu_2, pytest.mark.vllm, pytest.mark.h100], marks=[
pytest.mark.gpu_2,
pytest.mark.vllm,
pytest.mark.h100,
],
endpoints=["v1/chat/completions", "v1/completions"], endpoints=["v1/chat/completions", "v1/completions"],
response_handlers=[ response_handlers=[
chat_completions_response_handler, chat_completions_response_handler,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment