infra: handle flaky tests (#30501)

2025-09-15 22:44:36 +00:00 · 2025-03-26 13:28:56 -04:00
parent 9a80be7bb7
commit 422ba4cde5
6 changed files with 40 additions and 3 deletions
--- a/libs/partners/anthropic/tests/integration_tests/test_chat_models.py
+++ b/libs/partners/anthropic/tests/integration_tests/test_chat_models.py
@@ -730,6 +730,7 @@ def test_thinking() -> None:
            assert block["signature"] and isinstance(block["signature"], str)


+@pytest.mark.flaky(retries=3, delay=1)
 def test_redacted_thinking() -> None:
    llm = ChatAnthropic(
        model="claude-3-7-sonnet-latest",