diff --git a/src/dstack/_internal/server/migrations/versions/35e90e1b0d3e_add_rolling_deployment_fields.py b/src/dstack/_internal/server/migrations/versions/35e90e1b0d3e_add_rolling_deployment_fields.py index a0cac48af3..e26206222f 100644 --- a/src/dstack/_internal/server/migrations/versions/35e90e1b0d3e_add_rolling_deployment_fields.py +++ b/src/dstack/_internal/server/migrations/versions/35e90e1b0d3e_add_rolling_deployment_fields.py @@ -17,12 +17,6 @@ def upgrade() -> None: - with op.batch_alter_table("jobs", schema=None) as batch_op: - batch_op.add_column(sa.Column("deployment_num", sa.Integer(), nullable=True)) - with op.batch_alter_table("jobs", schema=None) as batch_op: - batch_op.execute("UPDATE jobs SET deployment_num = 0") - batch_op.alter_column("deployment_num", nullable=False) - with op.batch_alter_table("runs", schema=None) as batch_op: batch_op.add_column(sa.Column("deployment_num", sa.Integer(), nullable=True)) batch_op.add_column(sa.Column("desired_replica_count", sa.Integer(), nullable=True)) @@ -32,6 +26,12 @@ def upgrade() -> None: batch_op.alter_column("deployment_num", nullable=False) batch_op.alter_column("desired_replica_count", nullable=False) + with op.batch_alter_table("jobs", schema=None) as batch_op: + batch_op.add_column(sa.Column("deployment_num", sa.Integer(), nullable=True)) + with op.batch_alter_table("jobs", schema=None) as batch_op: + batch_op.execute("UPDATE jobs SET deployment_num = 0") + batch_op.alter_column("deployment_num", nullable=False) + def downgrade() -> None: with op.batch_alter_table("runs", schema=None) as batch_op: