From 2b8e2bd937221a18764867eb05960831f805f9a6 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 29 Dec 2023 10:52:45 +0530 Subject: [PATCH] (ci/cd) set num retries for HF test --- litellm/tests/test_text_completion.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_text_completion.py b/litellm/tests/test_text_completion.py index 30e87e67e27..4d4f021da44 100644 --- a/litellm/tests/test_text_completion.py +++ b/litellm/tests/test_text_completion.py @@ -19,6 +19,8 @@ from litellm import ( ) from litellm import RateLimitError +litellm.num_retries = 3 + token_prompt = [ [