test: cleanup tests

This commit is contained in:
Krrish Dholakia 2024-06-17 16:33:58 -07:00
parent 57d9479252
commit 1b215d704d
2 changed files with 17 additions and 9 deletions

View file

@ -3459,6 +3459,7 @@ def test_completion_palm_stream():
pytest.fail(f"Error occurred: {e}")
@pytest.mark.skip(reason="Account deleted by IBM.")
def test_completion_watsonx():
litellm.set_verbose = True
model_name = "watsonx/ibm/granite-13b-chat-v2"
@ -3479,6 +3480,7 @@ def test_completion_watsonx():
pytest.fail(f"Error occurred: {e}")
@pytest.mark.skip(reason="Skip test. account deleted.")
def test_completion_stream_watsonx():
litellm.set_verbose = True
model_name = "watsonx/ibm/granite-13b-chat-v2"