fix tests

mlcommons · Dec 10, 2024 · d6ddf71 · d6ddf71
1 parent 830e709
commit d6ddf71
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/plugins/mistral/tests/test_mistral_sut.py b/plugins/mistral/tests/test_mistral_sut.py
@@ -53,7 +53,7 @@ def response():
 
 @pytest.fixture
 def sut():
-    return MistralAISut("ministral-8b-latest", "ministral-8b-latest", "latest", MistralAIAPIKey("fake"))
+    return MistralAISut("ministral-8b-latest", "ministral-8b-latest", MistralAIAPIKey("fake"))
 
 
 @pytest.fixture
@@ -118,7 +118,7 @@ def moderated_response():
 @pytest.fixture
 def moderated_sut():
     return MistralAIModeratedSut(
-        "ministral-8b-latest", "ministral-8b-latest", "latest", "moderator", 2, 0.3, 0.3, MistralAIAPIKey("fake")
+        "ministral-8b-latest", "ministral-8b-latest", "moderator", 2, 0.3, 0.3, MistralAIAPIKey("fake")
     )
 
 
@@ -154,7 +154,7 @@ def test_response(self, moderated_sut, moderated_req, moderated_response):
 
     def test_response_over_safety_threshold(self, moderated_req, moderated_response):
         sut = MistralAIModeratedSut(
-            "ministral-8b-latest", "ministral-8b-latest", "latest", "moderator", 2, 0.3, 0.001, MistralAIAPIKey("fake")
+            "ministral-8b-latest", "ministral-8b-latest", "moderator", 2, 0.3, 0.001, MistralAIAPIKey("fake")
         )
         resp = sut.translate_response(request=moderated_req, response=moderated_response)
         assert resp == SUTResponse(completions=[SUTCompletion(text="I'm sorry I cannot assist with this request.")])