test(test_clarifai_completion.py): skip clarifai tests - account rate limited

This commit is contained in:
Krrish Dholakia 2024-06-03 21:06:58 -07:00
parent 0fa370ae44
commit 3d39d23f24

View file

@ -38,6 +38,7 @@ def reset_callbacks():
litellm.callbacks = [] litellm.callbacks = []
@pytest.mark.skip(reason="Account rate limited.")
def test_completion_clarifai_claude_2_1(): def test_completion_clarifai_claude_2_1():
print("calling clarifai claude completion") print("calling clarifai claude completion")
import os import os
@ -61,6 +62,7 @@ def test_completion_clarifai_claude_2_1():
pytest.fail(f"Error occured: {e}") pytest.fail(f"Error occured: {e}")
@pytest.mark.skip(reason="Account rate limited")
def test_completion_clarifai_mistral_large(): def test_completion_clarifai_mistral_large():
try: try:
litellm.set_verbose = True litellm.set_verbose = True
@ -80,6 +82,7 @@ def test_completion_clarifai_mistral_large():
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
@pytest.mark.skip(reason="Account rate limited")
@pytest.mark.asyncio @pytest.mark.asyncio
def test_async_completion_clarifai(): def test_async_completion_clarifai():
import asyncio import asyncio