From c63cef96b17c4879c3fa1dba31ddd1cbb9129f37 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 29 Aug 2023 18:23:34 -0700 Subject: [PATCH] updates --- litellm/tests/test_completion.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index fa3cf453de..59d414efff 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -348,13 +348,13 @@ def test_completion_together_ai(): pytest.fail(f"Error occurred: {e}") -def test_vertex_ai(): - model_name = "chat-bison" - try: - response = completion(model=model_name, messages=messages, logger_fn=logger_fn) - print(response) - except Exception as e: - pytest.fail(f"Error occurred: {e}") +# def test_vertex_ai(): +# model_name = "chat-bison" +# try: +# response = completion(model=model_name, messages=messages, logger_fn=logger_fn) +# print(response) +# except Exception as e: +# pytest.fail(f"Error occurred: {e}") def test_petals(): model_name = "stabilityai/StableBeluga2"