fixup! fix for sqlite

apache · Nov 22, 2024 · 5f5adbc · 5f5adbc
1 parent 7940140
commit 5f5adbc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/airflow/migrations/versions/0032_3_0_0_drop_execution_date_unique.py b/airflow/migrations/versions/0032_3_0_0_drop_execution_date_unique.py
@@ -31,9 +31,10 @@
 
 from __future__ import annotations
 
-import sqlalchemy as sa
 from alembic import op
 
+from airflow.utils.sqlalchemy import UtcDateTime
+
 # revision identifiers, used by Alembic.
 revision = "1cdc775ca98f"
 down_revision = "a2c32e6c7729"
@@ -47,7 +48,7 @@ def upgrade():
         batch_op.alter_column(
             "execution_date",
             new_column_name="logical_date",
-            existing_type=sa.TIMESTAMP,
+            existing_type=UtcDateTime,
             existing_nullable=False,
         )
     with op.batch_alter_table("dag_run", schema=None) as batch_op:
@@ -59,7 +60,7 @@ def downgrade():
         batch_op.alter_column(
             "logical_date",
             new_column_name="execution_date",
-            existing_type=sa.TIMESTAMP,
+            existing_type=UtcDateTime,
             existing_nullable=False,
         )
     with op.batch_alter_table("dag_run", schema=None) as batch_op:

diff --git a/docs/apache-airflow/img/airflow_erd.sha256 b/docs/apache-airflow/img/airflow_erd.sha256
@@ -1 +1 @@
-7173ce2b98bc53467274ebe89bd3d4c8e7672f7b40b3f22989139ece45741f8f
+a81bc0639419e66abacc09d95177df593b10d8944708212ea2a96a51070b7b31
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		7173ce2b98bc53467274ebe89bd3d4c8e7672f7b40b3f22989139ece45741f8f
		a81bc0639419e66abacc09d95177df593b10d8944708212ea2a96a51070b7b31