From eb7478a58a30b13c404e08c2da60ab95ba3d43a0 Mon Sep 17 00:00:00 2001
From: LawyZheng <lawyzheng1106@gmail.com>
Date: Mon, 1 Jul 2024 10:12:26 +0800
Subject: [PATCH] enable step created->failed (#534)

---
 skyvern/forge/sdk/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/skyvern/forge/sdk/models.py b/skyvern/forge/sdk/models.py
index a0bc39fdd..b4912e734 100644
--- a/skyvern/forge/sdk/models.py
+++ b/skyvern/forge/sdk/models.py
@@ -19,7 +19,7 @@ class StepStatus(StrEnum):
 
     def can_update_to(self, new_status: StepStatus) -> bool:
         allowed_transitions: dict[StepStatus, set[StepStatus]] = {
-            StepStatus.created: {StepStatus.running, StepStatus.canceled},
+            StepStatus.created: {StepStatus.running, StepStatus.failed, StepStatus.canceled},
             StepStatus.running: {StepStatus.completed, StepStatus.failed, StepStatus.canceled},
             StepStatus.failed: set(),
             StepStatus.completed: set(),