diff --git a/tests/llm_translation/base_llm_unit_tests.py b/tests/llm_translation/base_llm_unit_tests.py index d65735a6200..12df71839b2 100644 --- a/tests/llm_translation/base_llm_unit_tests.py +++ b/tests/llm_translation/base_llm_unit_tests.py @@ -347,6 +347,7 @@ class BaseLLMChatTest(ABC): assert response is not None print(f"response={response}") + @pytest.mark.flaky(retries=3, delay=1) @pytest.mark.parametrize("sync_mode", [True, False]) @pytest.mark.asyncio async def test_pdf_handling(self, pdf_messages, sync_mode):