(test) hf currently loading error

2024-01-15 17:07:49 -08:00 · 2024-01-15 17:07:49 -08:00 · 4e6a238820
commit 4e6a238820
parent 58a0b4c03b
1 changed files with 2 additions and 1 deletions
--- a/litellm/tests/test_text_completion.py
+++ b/litellm/tests/test_text_completion.py
@ -2833,8 +2833,9 @@ def test_completion_hf_prompt_array():
        assert len(response.choices) == 2
        # response_str = response["choices"][0]["text"]
    except Exception as e:
+        print(str(e))
        if "is currently loading" in str(e):
-            pass
+            return
        pytest.fail(f"Error occurred: {e}")