Bugfix for missing argument in Petals error handler (#237)

qcri · Oct 2, 2023 · 5843d54 · 5843d54
1 parent c4d6f6a
commit 5843d54
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmebench/models/Petals.py b/llmebench/models/Petals.py
@@ -127,6 +127,6 @@ def prompt(self, processed_input):
                 if not response["ok"]:
                     raise PetalsFailure("processing", response["traceback"])
             else:
-                raise PetalsFailure(connect_message["traceback"])
+                raise PetalsFailure("connection", connect_message["traceback"])
 
         return response