diff --git a/tests/integration/defs/accuracy/test_llm_api_pytorch.py b/tests/integration/defs/accuracy/test_llm_api_pytorch.py index d583408971..7ae3da0878 100644 --- a/tests/integration/defs/accuracy/test_llm_api_pytorch.py +++ b/tests/integration/defs/accuracy/test_llm_api_pytorch.py @@ -2916,7 +2916,8 @@ class TestGLM4_6(LlmapiAccuracyTestHarness): task.evaluate(llm) -@pytest.mark.timeout(7200) +@pytest.mark.threadleak(enabled=False) +@pytest.mark.timeout(10800) @pytest.mark.skip_less_device_memory(100000) class TestKimiK2(LlmapiAccuracyTestHarness): MODEL_NAME = "moonshotai/Kimi-K2-Instruct" @@ -2960,7 +2961,6 @@ class TestKimiK2(LlmapiAccuracyTestHarness): task.evaluate(llm) @skip_pre_blackwell - @pytest.mark.timeout(7200) @pytest.mark.skip_less_device_memory(120000) @pytest.mark.parametrize("tp_size", [ pytest.param(4, marks=pytest.mark.skip_less_device(4)),