From 5843d5493497707becf49803277acddf21174c3e Mon Sep 17 00:00:00 2001
From: Fahim Dalvi <faimaduddin@hbku.edu.qa>
Date: Mon, 2 Oct 2023 15:48:31 +0300
Subject: [PATCH] Bugfix for missing argument in Petals error handler (#237)

---
 llmebench/models/Petals.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/llmebench/models/Petals.py b/llmebench/models/Petals.py
index b25965e0..e0f6711d 100644
--- 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