diff --git a/tests/python_tests/continuous_batching/test_preemption.py b/tests/python_tests/continuous_batching/test_preemption.py index 0e935b6d4e..58abec9433 100644 --- a/tests/python_tests/continuous_batching/test_preemption.py +++ b/tests/python_tests/continuous_batching/test_preemption.py @@ -35,7 +35,7 @@ def test_preemption(tmp_path, params): # todo: Anastasiia Pnevskaya: fix the test because it is hanging according max_new_tokens = std::numeric_limits::max() @pytest.mark.parametrize("dynamic_split_fuse", [True, False]) @pytest.mark.precommit -@pytest.mark.xfail(raises=AssertionError, condition=sys.platform in ["win32", "darwin"]) +@pytest.mark.xfail(raises=AssertionError, reason="Fails on CI.", condition=sys.platform in ["win32", "darwin"]) def test_preemption_with_multinomial(tmp_path, dynamic_split_fuse): generation_configs = multinomial_params.generation_config for config in generation_configs: