From 48834c9b6340ca073146130ae583329e260db91a Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Fri, 22 Nov 2024 00:09:16 +0530
Subject: [PATCH] test: mark flaky test

---
 tests/llm_translation/base_llm_unit_tests.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/llm_translation/base_llm_unit_tests.py b/tests/llm_translation/base_llm_unit_tests.py
index 354151bd8..88fce6dac 100644
--- a/tests/llm_translation/base_llm_unit_tests.py
+++ b/tests/llm_translation/base_llm_unit_tests.py
@@ -92,6 +92,7 @@ class BaseLLMChatTest(ABC):
         # relevant issue: https://github.com/BerriAI/litellm/issues/6741
         assert response.choices[0].message.content is not None
 
+    @pytest.mark.flaky(retries=6, delay=1)
     def test_json_response_pydantic_obj(self):
         litellm.set_verbose = True
         from pydantic import BaseModel