diff --git a/tests/llm/test_openai/evals/test_sentiment_analysis.py b/tests/llm/test_openai/evals/test_sentiment_analysis.py index 1b156db0a..303e1dbd9 100644 --- a/tests/llm/test_openai/evals/test_sentiment_analysis.py +++ b/tests/llm/test_openai/evals/test_sentiment_analysis.py @@ -18,9 +18,18 @@ class SentimentAnalysis(BaseModel): test_data = [ - ("I absolutely love this product! It has exceeded all my expectations.", Sentiment.POSITIVE), - ("The service was terrible. I will never use this company again.", Sentiment.NEGATIVE), - ("The movie was okay. It had some good moments but overall it was average.", Sentiment.NEUTRAL), + ( + "I absolutely love this product! It has exceeded all my expectations.", + Sentiment.POSITIVE, + ), + ( + "The service was terrible. I will never use this company again.", + Sentiment.NEGATIVE, + ), + ( + "The movie was okay. It had some good moments but overall it was average.", + Sentiment.NEUTRAL, + ), ] @@ -48,4 +57,4 @@ def test_sentiment_analysis(model, data, mode, client): ], ) - assert response.sentiment == expected_sentiment \ No newline at end of file + assert response.sentiment == expected_sentiment diff --git a/tests/test_new_client.py b/tests/test_new_client.py index 50af0893b..e66e0652e 100644 --- a/tests/test_new_client.py +++ b/tests/test_new_client.py @@ -342,6 +342,7 @@ class Group(BaseModel): @pytest.mark.skip(reason="Skip for now") def test_client_from_mistral_with_response(): import mistralai.client as mistralaicli + client = instructor.from_mistral( mistralaicli.MistralClient(), max_tokens=1000, @@ -360,6 +361,7 @@ def test_client_from_mistral_with_response(): @pytest.mark.skip(reason="Skip for now") def test_client_mistral_response(): import mistralai.client as mistralaicli + client = mistralaicli.MistralClient() instructor_client = instructor.from_mistral( client, max_tokens=1000, model="mistral-large-latest"