Skip to content

Commit e4a41f9

Browse files
krishung5hhzhang16
authored andcommitted
fix: Update pytest markers for deepep test. (#2561)
Signed-off-by: Hannah Zhang <hannahz@nvidia.com>
1 parent 0de05cc commit e4a41f9

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

components/backends/vllm/launch/dsr1_dep.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,8 +47,8 @@ while [[ $# -gt 0 ]]; do
4747
echo " --gpus-per-node L Number of GPUs per node (required, int)"
4848
echo " --master-addr ADDR Master node address (default: localhost)"
4949
echo " --log-dir DIR Directory for log files (default: ./logs)"
50-
echo " --model MODEL Model name to use (default: deepseek-ai/DeepSeek-R1)"
51-
echo " -h, --help Show this help message"
50+
echo " --model MODEL Model name to use (default: ${MODEL})"
51+
echo " -h, --help Show this help message"
5252
exit 0
5353
;;
5454
*)

tests/serve/test_vllm.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -264,7 +264,11 @@ def wait_for_ready(self, payload, logger=logging.getLogger()):
264264
name="deepep",
265265
directory="/workspace/components/backends/vllm",
266266
script_name="dsr1_dep.sh",
267-
marks=[pytest.mark.gpu_2, pytest.mark.vllm, pytest.mark.h100],
267+
marks=[
268+
pytest.mark.gpu_2,
269+
pytest.mark.vllm,
270+
pytest.mark.h100,
271+
],
268272
endpoints=["v1/chat/completions", "v1/completions"],
269273
response_handlers=[
270274
chat_completions_response_handler,

0 commit comments

Comments
 (0)