From 1bd92b20dd49370b412da25f9a4549f5f265ce5d Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 4 Mar 2024 08:44:39 -0800 Subject: [PATCH] (test-fix) hf is unstable --- 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 0b981118397..a16b864494b 100644 --- a/litellm/tests/test_text_completion.py +++ b/litellm/tests/test_text_completion.py @@ -2836,6 +2836,8 @@ def test_completion_hf_prompt_array(): print(str(e)) if "is currently loading" in str(e): return + if "Service Unavailable" in str(e): + return pytest.fail(f"Error occurred: {e}")