Skip to content

Commit 6994b30

Browse files
committed
Update Mistral and TogetherAI tests [release]
1 parent 8e546d2 commit 6994b30

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

src/tests/llms/test_llms.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import multiprocessing
55
import os
66
import sys
7+
import typing
78
import uuid
89
from collections import Counter, defaultdict
910
from concurrent.futures import ThreadPoolExecutor
@@ -2988,6 +2989,7 @@ def test_get_max_context_length(self, create_datadreamer):
29882989
)
29892990

29902991
@pytest.mark.order("last")
2992+
@typing.no_type_check
29912993
def test_run(self, create_datadreamer, mocker):
29922994
def chat_mocked(**kwargs):
29932995
from mistralai.models.chat_completion import ChatCompletionResponse
@@ -3014,7 +3016,7 @@ def chat_mocked(**kwargs):
30143016
"total_tokens": 0,
30153017
},
30163018
}
3017-
return ChatCompletionResponse(**response) # type: ignore[arg-type]
3019+
return ChatCompletionResponse(**response)
30183020

30193021
with create_datadreamer():
30203022
llm = MistralAI("mistral-tiny", api_key="fakeapikey")

0 commit comments

Comments
 (0)