(airflow) branch main updated: Use canonical exception paths in task sdk (#60700)

[shahar]

This is an automated email from the ASF dual-hosted git repository.

shahar pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/main by this push:
     new b9ad6567af3 Use canonical exception paths in task sdk (#60700)
b9ad6567af3 is described below

commit b9ad6567af323de7715e2513571535b3d4cbb292
Author: Amogh Desai <amoghrajesh1...@gmail.com>
AuthorDate: Sat Jan 17 19:24:44 2026 +0530

    Use canonical exception paths in task sdk (#60700)
---
 task-sdk/src/airflow/sdk/execution_time/task_runner.py     | 2 +-
 task-sdk/tests/task_sdk/definitions/test_taskgroup.py      | 4 ++--
 task-sdk/tests/task_sdk/execution_time/test_task_runner.py | 4 ++--
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/task-sdk/src/airflow/sdk/execution_time/task_runner.py 
b/task-sdk/src/airflow/sdk/execution_time/task_runner.py
index 95665c43f3b..60577066998 100644
--- a/task-sdk/src/airflow/sdk/execution_time/task_runner.py
+++ b/task-sdk/src/airflow/sdk/execution_time/task_runner.py
@@ -765,7 +765,7 @@ def _verify_bundle_access(bundle_instance: BaseDagBundle, 
log: Logger) -> None:
     """
     from getpass import getuser
 
-    from airflow.exceptions import AirflowException
+    from airflow.sdk.exceptions import AirflowException
 
     bundle_path = bundle_instance.path
 
diff --git a/task-sdk/tests/task_sdk/definitions/test_taskgroup.py 
b/task-sdk/tests/task_sdk/definitions/test_taskgroup.py
index 4c2b00eb2ac..6eb51a1a2f6 100644
--- a/task-sdk/tests/task_sdk/definitions/test_taskgroup.py
+++ b/task-sdk/tests/task_sdk/definitions/test_taskgroup.py
@@ -20,7 +20,6 @@ from __future__ import annotations
 import pendulum
 import pytest
 
-from airflow.exceptions import TaskAlreadyInTaskGroup
 from airflow.sdk import (
     DAG,
     XComArg,
@@ -32,6 +31,7 @@ from airflow.sdk import (
     timezone,
 )
 from airflow.sdk.definitions.taskgroup import TaskGroup
+from airflow.sdk.exceptions import TaskAlreadyInTaskGroup
 
 from tests_common.test_utils.compat import BashOperator, EmptyOperator, 
PythonOperator
 
@@ -426,7 +426,7 @@ def test_dag_edges_task_group_structure():
 
 
 def test_duplicate_group_id():
-    from airflow.exceptions import DuplicateTaskIdFound
+    from airflow.sdk.exceptions import DuplicateTaskIdFound
 
     logical_date = pendulum.parse("20200101")
 
diff --git a/task-sdk/tests/task_sdk/execution_time/test_task_runner.py 
b/task-sdk/tests/task_sdk/execution_time/test_task_runner.py
index f743abdd9a6..9cf63c654aa 100644
--- a/task-sdk/tests/task_sdk/execution_time/test_task_runner.py
+++ b/task-sdk/tests/task_sdk/execution_time/test_task_runner.py
@@ -2949,7 +2949,7 @@ class TestEmailNotifications:
     )
     def test_email_on_failure(self, emails, sent, create_runtime_ti, 
mock_supervisor_comms):
         """Test email notification on task failure."""
-        from airflow.exceptions import AirflowFailException
+        from airflow.sdk.exceptions import AirflowFailException
         from airflow.sdk.execution_time.task_runner import finalize, run
 
         class FailingOperator(BaseOperator):
@@ -2985,7 +2985,7 @@ class TestEmailNotifications:
 
     def test_email_with_custom_templates(self, create_runtime_ti, 
mock_supervisor_comms, tmp_path):
         """Test email notification respects custom subject and html_content 
templates."""
-        from airflow.exceptions import AirflowFailException
+        from airflow.sdk.exceptions import AirflowFailException
 
         subject_template = tmp_path / "custom_subject.jinja2"
         html_template = tmp_path / "custom_html.html"