From fab8d264b096c4ced0785054388700b3d9c9330f Mon Sep 17 00:00:00 2001 From: wangli Date: Fri, 16 May 2025 10:21:35 +0800 Subject: [PATCH 1/3] add upper time to distributed test Signed-off-by: wangli --- requirements-dev.txt | 1 + tests/multicard/test_offline_inference_distributed.py | 1 + 2 files changed, 2 insertions(+) diff --git a/requirements-dev.txt b/requirements-dev.txt index 4fb45d11b..3f8bd086a 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -3,6 +3,7 @@ modelscope openai pytest >= 6.0 +pytest-timeout pytest-asyncio lm-eval ray diff --git a/tests/multicard/test_offline_inference_distributed.py b/tests/multicard/test_offline_inference_distributed.py index 26a3de5ac..e0e16b8bd 100644 --- a/tests/multicard/test_offline_inference_distributed.py +++ b/tests/multicard/test_offline_inference_distributed.py @@ -35,6 +35,7 @@ ("Qwen/QwQ-32B", "mp"), ("deepseek-ai/DeepSeek-V2-Lite", "mp"), ]) +@pytest.mark.timeout(10*60*60) def test_models_distributed(model: str, distributed_executor_backend: str) -> None: example_prompts = [ From 384a07942df020bd4222b1e8861025bee893689b Mon Sep 17 00:00:00 2001 From: wangli Date: Fri, 16 May 2025 10:25:15 +0800 Subject: [PATCH 2/3] fix yapf Signed-off-by: wangli --- tests/multicard/test_offline_inference_distributed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/multicard/test_offline_inference_distributed.py b/tests/multicard/test_offline_inference_distributed.py index e0e16b8bd..359a386f3 100644 --- a/tests/multicard/test_offline_inference_distributed.py +++ b/tests/multicard/test_offline_inference_distributed.py @@ -35,7 +35,7 @@ ("Qwen/QwQ-32B", "mp"), ("deepseek-ai/DeepSeek-V2-Lite", "mp"), ]) -@pytest.mark.timeout(10*60*60) +@pytest.mark.timeout(10 * 60 * 60) def test_models_distributed(model: str, distributed_executor_backend: str) -> None: example_prompts = [ From 2e87d9c79cda481fde9bd5bccee60d255aebc854 Mon Sep 17 00:00:00 2001 From: wangli Date: Fri, 16 May 2025 10:27:16 +0800 Subject: [PATCH 3/3] change timeou time Signed-off-by: wangli --- tests/multicard/test_offline_inference_distributed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/multicard/test_offline_inference_distributed.py b/tests/multicard/test_offline_inference_distributed.py index 359a386f3..5de98256d 100644 --- a/tests/multicard/test_offline_inference_distributed.py +++ b/tests/multicard/test_offline_inference_distributed.py @@ -35,7 +35,7 @@ ("Qwen/QwQ-32B", "mp"), ("deepseek-ai/DeepSeek-V2-Lite", "mp"), ]) -@pytest.mark.timeout(10 * 60 * 60) +@pytest.mark.timeout(15 * 60) def test_models_distributed(model: str, distributed_executor_backend: str) -> None: example_prompts = [