From b13618072859c3cc5f89c63bda2d4a9a238764aa Mon Sep 17 00:00:00 2001
From: Amogh Desai <amoghrajesh1999@gmail.com>
Date: Sat, 22 Mar 2025 08:50:14 +0530
Subject: [PATCH] Fixing CI in main due to failing generate sdk datamodels

---
 .../airflow/sdk/api/datamodels/_generated.py  | 20 +++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/task-sdk/src/airflow/sdk/api/datamodels/_generated.py b/task-sdk/src/airflow/sdk/api/datamodels/_generated.py
index 0ed6fc985becd..2fb929516cfbb 100644
--- a/task-sdk/src/airflow/sdk/api/datamodels/_generated.py
+++ b/task-sdk/src/airflow/sdk/api/datamodels/_generated.py
@@ -97,6 +97,14 @@ class DagRunAssetReference(BaseModel):
 
 
 class DagRunState(str, Enum):
+    """
+    All possible states that a DagRun can be in.
+
+    These are "shared" with TaskInstanceState in some parts of the code,
+    so please ensure that their values always match the ones with the
+    same name in TaskInstanceState.
+    """
+
     QUEUED = "queued"
     RUNNING = "running"
     SUCCESS = "success"
@@ -112,6 +120,10 @@ class DagRunStateResponse(BaseModel):
 
 
 class DagRunType(str, Enum):
+    """
+    Class with DagRun types.
+    """
+
     BACKFILL = "backfill"
     SCHEDULED = "scheduled"
     MANUAL = "manual"
@@ -119,6 +131,10 @@ class DagRunType(str, Enum):
 
 
 class IntermediateTIState(str, Enum):
+    """
+    States that a Task Instance can be in that indicate it is not yet in a terminal or running state.
+    """
+
     SCHEDULED = "scheduled"
     QUEUED = "queued"
     RESTARTING = "restarting"
@@ -256,6 +272,10 @@ class TITargetStatePayload(BaseModel):
 
 
 class TerminalStateNonSuccess(str, Enum):
+    """
+    TaskInstance states that can be reported without extra information.
+    """
+
     FAILED = "failed"
     SKIPPED = "skipped"
     REMOVED = "removed"