code: def upgrade() -> None:
      # ### commands auto generated by Alembic - please adjust! ###
      with op.batch_alter_table("jobs", schema=None) as batch_op:
          batch_op.add_column(
              sa.Column(
                  "removed",
                  sa.Boolean(),
                  nullable=False,
                  server_default="0",
              )
          )
          batch_op.add_column(
              sa.Column("remove_at", sa.DateTime(), nullable=True)
          )
  
      # ### end Alembic commands ###
      op.execute(
          sa.sql.text(
              "UPDATE jobs SET removed = TRUE WHERE status IN :statuses"
          ).bindparams(
              sa.bindparam("statuses", expanding=True),
              statuses=tuple(
                  i.value.upper() for i in JobStatus.finished_statuses()
              ),
          )
      )
  
imports: import sqlalchemy
  import sqlalchemy_utils
  from alembic import op
  from dstack._internal.core.models.runs import JobStatus
label_desc: Use image node_id 1 for calling a global function with example usage: upgrade() without return types