Fix SQLA Mapped type import errors in model classes

airflow-core/src/airflow/models/dag_favorite.py

@@ -17,16 +17,12 @@
 # under the License.
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
-
 from sqlalchemy import ForeignKey
+from sqlalchemy.orm import Mapped  # noqa: TC002
 
 from airflow.models.base import Base, StringID
 from airflow.utils.sqlalchemy import mapped_column
 
-if TYPE_CHECKING:
-    from sqlalchemy.orm import Mapped
-
 
 class DagFavorite(Base):
     """Association table model linking users to their favorite DAGs."""

airflow-core/src/airflow/models/db_callback_request.py

@@ -21,14 +21,13 @@
 from typing import TYPE_CHECKING
 
 from sqlalchemy import Integer, String
+from sqlalchemy.orm import Mapped  # noqa: TC002
 
 from airflow._shared.timezones import timezone
 from airflow.models.base import Base
 from airflow.utils.sqlalchemy import ExtendedJSON, UtcDateTime, mapped_column
 
 if TYPE_CHECKING:
-    from sqlalchemy.orm import Mapped
-
     from airflow.callbacks.callback_requests import CallbackRequest
 
 

airflow-core/src/airflow/models/errors.py

@@ -17,17 +17,13 @@
 # under the License.
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
-
 from sqlalchemy import Integer, String, Text
+from sqlalchemy.orm import Mapped  # noqa: TC002
 
 from airflow.dag_processing.bundles.manager import DagBundlesManager
 from airflow.models.base import Base, StringID
 from airflow.utils.sqlalchemy import UtcDateTime, mapped_column
 
-if TYPE_CHECKING:
-    from sqlalchemy.orm import Mapped
-
 
 class ParseImportError(Base):
     """Stores all Import Errors which are recorded when parsing DAGs and displayed on the Webserver."""

airflow-core/src/airflow/models/pool.py

@@ -20,6 +20,7 @@
 from typing import TYPE_CHECKING, Any, TypedDict
 
 from sqlalchemy import Boolean, ForeignKey, Integer, String, Text, func, select
+from sqlalchemy.orm import Mapped  # noqa: TC002
 from sqlalchemy_utils import UUIDType
 
 from airflow.exceptions import AirflowException, PoolNotFound
@@ -32,7 +33,6 @@
 from airflow.utils.state import TaskInstanceState
 
 if TYPE_CHECKING:
-    from sqlalchemy.orm import Mapped
     from sqlalchemy.orm.session import Session
 
 

airflow-core/src/airflow/models/tasklog.py

@@ -17,17 +17,13 @@
 # under the License.
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
-
 from sqlalchemy import Integer, Text
+from sqlalchemy.orm import Mapped  # noqa: TC002
 
 from airflow._shared.timezones import timezone
 from airflow.models.base import Base
 from airflow.utils.sqlalchemy import UtcDateTime, mapped_column
 
-if TYPE_CHECKING:
-    from sqlalchemy.orm import Mapped
-
 
 class LogTemplate(Base):
     """

airflow-core/src/airflow/models/taskmap.py

@@ -25,6 +25,7 @@
 from typing import TYPE_CHECKING, Any
 
 from sqlalchemy import CheckConstraint, ForeignKeyConstraint, Integer, String, func, or_, select
+from sqlalchemy.orm import Mapped  # noqa: TC002
 
 from airflow.models.base import COLLATION_ARGS, ID_LEN, TaskInstanceDependencies
 from airflow.models.dag_version import DagVersion
@@ -33,7 +34,7 @@
 from airflow.utils.state import State, TaskInstanceState
 
 if TYPE_CHECKING:
-    from sqlalchemy.orm import Mapped, Session
+    from sqlalchemy.orm import Session
 
     from airflow.models.mappedoperator import MappedOperator
     from airflow.models.taskinstance import TaskInstance