move TaskNotFound to sdk

ephraimbuddy · ephraimbuddy · commit e2df3c1bd1de · 2025-10-30T13:03:59.000+01:00
diff --git a/airflow-core/src/airflow/dag_processing/processor.py b/airflow-core/src/airflow/dag_processing/processor.py
@@ -36,7 +36,7 @@
 )
 from airflow.configuration import conf
 from airflow.dag_processing.dagbag import DagBag
-from airflow.exceptions import TaskNotFound
+from airflow.sdk.exceptions import TaskNotFound
 from airflow.sdk.execution_time.comms import (
     ConnectionResult,
     DeleteVariable,
diff --git a/airflow-core/src/airflow/exceptions.py b/airflow-core/src/airflow/exceptions.py
@@ -311,7 +311,6 @@ def __init__(self, dag_id: str | None = None, message: str | None = None):
     "AirflowSkipException": "airflow.sdk.exceptions.AirflowSkipException",
     "AirflowTaskTimeout": "airflow.sdk.exceptions.AirflowTaskTimeout",
     "AirflowFailException": "airflow.sdk.exceptions.AirflowFailException",
-    # TODO: This will be handled as part of https://github.com/apache/airflow/pull/55111
     "ParamValidationError": "airflow.sdk.exceptions.ParamValidationError",
     "TaskDeferred": "airflow.sdk.exceptions.TaskDeferred",
 }

Original file line number	Diff line number	Diff line change
`@@ -36,7 +36,7 @@`
`36`	`36`	`)`
`37`	`37`	`from airflow.configuration import conf`
`38`	`38`	`from airflow.dag_processing.dagbag import DagBag`
`39`		`-from airflow.exceptions import TaskNotFound`
	`39`	`+from airflow.sdk.exceptions import TaskNotFound`
`40`	`40`	`from airflow.sdk.execution_time.comms import (`
`41`	`41`	`ConnectionResult,`
`42`	`42`	`DeleteVariable,`
Original file line number	Diff line number	Diff line change
`@@ -311,7 +311,6 @@ def __init__(self, dag_id: str \| None = None, message: str \| None = None):`
`311`	`311`	`"AirflowSkipException": "airflow.sdk.exceptions.AirflowSkipException",`
`312`	`312`	`"AirflowTaskTimeout": "airflow.sdk.exceptions.AirflowTaskTimeout",`
`313`	`313`	`"AirflowFailException": "airflow.sdk.exceptions.AirflowFailException",`
`314`		`- # TODO: This will be handled as part of https://github.com/apache/airflow/pull/55111`
`315`	`314`	`"ParamValidationError": "airflow.sdk.exceptions.ParamValidationError",`
`316`	`315`	`"TaskDeferred": "airflow.sdk.exceptions.TaskDeferred",`
`317`	`316`	`}`