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..5de98256d 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(15 * 60) def test_models_distributed(model: str, distributed_executor_backend: str) -> None: example_prompts = [