Fixing downgrade script for warning_text

apache · Sep 22, 2017 · 94f22be · 94f22be
1 parent 544f681
commit 94f22be
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/superset/migrations/versions/19a814813610_adding_metric_warning_text.py b/superset/migrations/versions/19a814813610_adding_metric_warning_text.py
@@ -15,11 +15,12 @@
 
 
 def upgrade():
-    # ### commands auto generated by Alembic - please adjust! ###
     op.add_column('metrics', sa.Column('warning_text', sa.Text(), nullable=True))
     op.add_column('sql_metrics', sa.Column('warning_text', sa.Text(), nullable=True))
 
 
 def downgrade():
-    op.drop_column('sql_metrics', 'warning_text')
-    op.drop_column('metrics', 'warning_text')
+    with op.batch_alter_table('sql_metrics') as batch_op_sql_metrics:
+        batch_op_sql_metrics.drop_column('warning_text')
+    with op.batch_alter_table('metrics') as batch_op_metrics:
+        batch_op_metrics.drop_column('warning_text')