From 35fed9aa4d6de260ec55ab5558ce75916f74a054 Mon Sep 17 00:00:00 2001
From: bennykok <itechbenny@gmail.com>
Date: Sat, 20 Apr 2024 01:38:31 +0800
Subject: [PATCH] fix: failed case marked as success

---
 custom_routes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/custom_routes.py b/custom_routes.py
index 4a4ac16..d83910b 100644
--- a/custom_routes.py
+++ b/custom_routes.py
@@ -717,7 +717,7 @@ def update_run(prompt_id: str, status: Status):
     if (prompt_metadata[prompt_id].status != status):
 
         # when the status is already failed, we don't want to update it to success
-        if ('status' in prompt_metadata[prompt_id] and prompt_metadata[prompt_id].status == Status.FAILED):
+        if (prompt_metadata[prompt_id].status is Status.FAILED):
             return
 
         status_endpoint = prompt_metadata[prompt_id].status_endpoint