oops

mlcommons · Dec 10, 2024 · 12530f5 · 12530f5
1 parent 8282f8b
commit 12530f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/plugins/mistral/modelgauge/suts/mistral_client.py b/plugins/mistral/modelgauge/suts/mistral_client.py
@@ -68,7 +68,7 @@ def request(self, req: dict):
         if self.client.chat.sdk_configuration._hooks.before_request_hooks:
             # work around bug in client
             self.client.chat.sdk_configuration._hooks.before_request_hooks = []
-        return self._retry_request(self.client.chat.complete, req)
+        return self._make_request(self.client.chat.complete, req)
 
     def score_conversation(self, model, prompt, response):
         """Returns moderation object for a conversation."""
@@ -79,4 +79,4 @@ def score_conversation(self, model, prompt, response):
                 {"role": "assistant", "content": response},
             ],
         }
-        return self._retry_request(self.client.classifiers.moderate_chat, req)
+        return self._make_request(self.client.classifiers.moderate_chat, req)