diff --git a/skyvern/forge/sdk/schemas/tasks.py b/skyvern/forge/sdk/schemas/tasks.py index 5051944f..b5dab4ea 100644 --- a/skyvern/forge/sdk/schemas/tasks.py +++ b/skyvern/forge/sdk/schemas/tasks.py @@ -106,6 +106,7 @@ def can_update_to(self, new_status: TaskStatus) -> bool: TaskStatus.failed: set(), TaskStatus.terminated: set(), TaskStatus.completed: set(), + TaskStatus.timed_out: set(), } return new_status in allowed_transitions[self]