refactor(taskinstance): change the exception type of "no exception and validate start_trigger_args passed to _defer_task" to TaskDeferralError

Lee-W · Lee-W · commit 636265e11961 · 2024-07-22T15:49:03.000+08:00
diff --git a/airflow/models/taskinstance.py b/airflow/models/taskinstance.py
@@ -1634,7 +1634,7 @@ def _defer_task(
             kwargs=trigger_kwargs,
         )
     else:
-        raise AirflowException("exception and ti.task.start_trigger_args cannot both be None")
+        raise TaskDeferralError("exception and ti.task.start_trigger_args cannot both be None")
 
     # First, make the trigger entry
     session.add(trigger_row)

Original file line number	Diff line number	Diff line change
`@@ -1634,7 +1634,7 @@ def _defer_task(`
`1634`	`1634`	`kwargs=trigger_kwargs,`
`1635`	`1635`	`)`
`1636`	`1636`	`else:`
`1637`		`- raise AirflowException("exception and ti.task.start_trigger_args cannot both be None")`
	`1637`	`+ raise TaskDeferralError("exception and ti.task.start_trigger_args cannot both be None")`
`1638`	`1638`
`1639`	`1639`	`# First, make the trigger entry`
`1640`	`1640`	`session.add(trigger_row)`