FIX-OCT-1576: fix altering migration

golemfoundation · May 28, 2024 · 07ff61f · 07ff61f
1 parent 7477ca7
commit 07ff61f
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/backend/migrations/versions/58536423cfdd_unify_all_individual_rewards_to_vanilla_.py b/backend/migrations/versions/58536423cfdd_unify_all_individual_rewards_to_vanilla_.py
@@ -16,15 +16,15 @@
 
 def upgrade():
     with op.batch_alter_table("pending_epoch_snapshots", schema=None) as batch_op:
-        batch_op.add_column(
-            sa.Column("vanilla_individual_rewards", sa.String(), nullable=False)
+        batch_op.alter_column(
+            column_name="all_individual_rewards",
+            new_column_name="vanilla_individual_rewards",
         )
-        batch_op.drop_column("all_individual_rewards")
 
 
 def downgrade():
     with op.batch_alter_table("pending_epoch_snapshots", schema=None) as batch_op:
-        batch_op.add_column(
-            sa.Column("all_individual_rewards", sa.VARCHAR(), nullable=False)
+        batch_op.alter_column(
+            column_name="vanilla_individual_rewards",
+            new_column_name="all_individual_rewards",
         )
-        batch_op.drop_column("vanilla_individual_rewards")