From 6994b30cb4fba0d153067e8b82841967d6e33f4a Mon Sep 17 00:00:00 2001 From: Ajay Patel Date: Fri, 2 Aug 2024 15:15:58 -0400 Subject: [PATCH] Update Mistral and TogetherAI tests [release] --- src/tests/llms/test_llms.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/tests/llms/test_llms.py b/src/tests/llms/test_llms.py index cb67eda..b7d62dc 100644 --- a/src/tests/llms/test_llms.py +++ b/src/tests/llms/test_llms.py @@ -4,6 +4,7 @@ import multiprocessing import os import sys +import typing import uuid from collections import Counter, defaultdict from concurrent.futures import ThreadPoolExecutor @@ -2988,6 +2989,7 @@ def test_get_max_context_length(self, create_datadreamer): ) @pytest.mark.order("last") + @typing.no_type_check def test_run(self, create_datadreamer, mocker): def chat_mocked(**kwargs): from mistralai.models.chat_completion import ChatCompletionResponse @@ -3014,7 +3016,7 @@ def chat_mocked(**kwargs): "total_tokens": 0, }, } - return ChatCompletionResponse(**response) # type: ignore[arg-type] + return ChatCompletionResponse(**response) with create_datadreamer(): llm = MistralAI("mistral-tiny", api_key="fakeapikey")