From 9f18aa4588ddcdb170d5a1e5caea43c434749823 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 30 Aug 2024 15:57:44 -0700 Subject: [PATCH] fix test_vertexai_embedding_embedding_latest_input_type --- litellm/tests/test_amazing_vertex_completion.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index dea10fe9bf2..4b659944a0a 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -2014,9 +2014,7 @@ def test_vertexai_embedding_embedding_latest(): pytest.fail(f"Error occurred: {e}") -# @pytest.mark.skip( -# reason="new test - works locally running into vertex version issues on ci/cd" -# ) +@pytest.mark.flaky(retries=3, delay=1) def test_vertexai_embedding_embedding_latest_input_type(): try: load_vertex_ai_credentials() @@ -2027,8 +2025,6 @@ def test_vertexai_embedding_embedding_latest_input_type(): input=["hi"], input_type="RETRIEVAL_QUERY", ) - - assert len(response.data[0]["embedding"]) == 1 assert response.usage.prompt_tokens > 0 print(f"response:", response) except litellm.RateLimitError as e: