Don't batch migrations if we don't need to

apache · Nov 25, 2022 · 652f745 · 652f745
1 parent 2cb2d99
commit 652f745
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/airflow/migrations/versions/0122_2_5_0_add_is_orphaned_to_datasetmodel.py b/airflow/migrations/versions/0122_2_5_0_add_is_orphaned_to_datasetmodel.py
@@ -39,11 +39,9 @@
 
 def upgrade():
     """Add is_orphaned to DatasetModel"""
-    with op.batch_alter_table("dataset") as batch_op:
-        batch_op.add_column(sa.Column("is_orphaned", sa.Boolean, nullable=False, server_default="False"))
+    op.add_column("dataset", sa.Column("is_orphaned", sa.Boolean, nullable=False, server_default="False"))
 
 
 def downgrade():
-    """Remove is_orphaned to DatasetModel"""
-    with op.batch_alter_table("dataset", schema=None) as batch_op:
-        batch_op.drop_column("is_orphaned")
+    """Remove is_orphaned from DatasetModel"""
+    op.drop_column("dataset", "is_orphaned")
diff --git a/docs/apache-airflow/img/airflow_erd.sha256 b/docs/apache-airflow/img/airflow_erd.sha256
@@ -1 +1 @@
-ab67d584c9912231e6963abdd710ede4228cb2f23d67219485361c75d4f9bbd7
+7854eceb2ff93010d984e5485c4266f6600eccab934f1aa86900b8b58bfc0915