mistral lint

UKGovernmentBEIS · Sep 19, 2024 · f6c4b73 · f6c4b73
1 parent 6ad50da
commit f6c4b73
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/inspect_ai/model/_providers/mistral.py b/src/inspect_ai/model/_providers/mistral.py
@@ -3,7 +3,6 @@
 from typing import Any
 
 from mistralai import (
-    ChatCompletionRequestToolChoice,
     FunctionCall,
     FunctionName,
     Mistral,

diff --git a/tests/tools/test_tools.py b/tests/tools/test_tools.py
@@ -77,7 +77,9 @@ async def add(x: int, y: int):
     return add
 
 
-def check_tools(model: Model, disable: list[str] = ["calls", "force", "none"]) -> None:
+def check_tools(
+    model: Model, disable: list[Literal["calls", "force", "none"]] = []
+) -> None:
     if "calls" not in disable:
         check_tools_calls(model)
     if "force" not in disable:
@@ -171,7 +173,7 @@ def test_anthropic_tools():
 
 @skip_if_no_mistral
 def test_mistral_tools():
-    check_tools("mistral/mistral-large-latest", disable=["force"])
+    check_tools("mistral/mistral-large-latest")
 
 
 @skip_if_no_groq