From 046b5267c96fc394f32456ccdd17597a2f371132 Mon Sep 17 00:00:00 2001 From: Amogh Desai Date: Tue, 30 Sep 2025 22:37:49 +0530 Subject: [PATCH] Fix SQLA Mapped datetime imports in models --- airflow-core/src/airflow/models/backfill.py | 3 +-- airflow-core/src/airflow/models/dagrun.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/airflow-core/src/airflow/models/backfill.py b/airflow-core/src/airflow/models/backfill.py index d3ff17f932edb..a9d0f2b3e344d 100644 --- a/airflow-core/src/airflow/models/backfill.py +++ b/airflow-core/src/airflow/models/backfill.py @@ -24,6 +24,7 @@ from __future__ import annotations import logging +from datetime import datetime from enum import Enum from typing import TYPE_CHECKING @@ -51,8 +52,6 @@ from airflow.utils.types import DagRunTriggeredByType, DagRunType if TYPE_CHECKING: - from datetime import datetime - from airflow.serialization.serialized_objects import SerializedDAG from airflow.timetables.base import DagRunInfo diff --git a/airflow-core/src/airflow/models/dagrun.py b/airflow-core/src/airflow/models/dagrun.py index 88b577e806ec1..0c8ff67d2312b 100644 --- a/airflow-core/src/airflow/models/dagrun.py +++ b/airflow-core/src/airflow/models/dagrun.py @@ -22,6 +22,7 @@ import re from collections import defaultdict from collections.abc import Callable, Iterable, Iterator, Sequence +from datetime import datetime from typing import TYPE_CHECKING, Any, NamedTuple, TypeVar, cast, overload import structlog @@ -85,7 +86,6 @@ from airflow.utils.types import NOTSET, DagRunTriggeredByType, DagRunType if TYPE_CHECKING: - from datetime import datetime from typing import Literal, TypeAlias from opentelemetry.sdk.trace import Span