"""empty message

Revision ID: 0998f67d7bd4
Revises: b73b0d3b5be7
Create Date: 2024-09-11 11:22:58.559355

"""
from alembic import op
import sqlalchemy as sa
import sqlmodel.sql.sqltypes
from tidb_vector.sqlalchemy import VectorType


# revision identifiers, used by Alembic.
revision = '0998f67d7bd4'
down_revision = 'b73b0d3b5be7'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('regression_dataset', sa.Column('status', sa.Enum('AVAILABLE', 'ARCHIVED', name='samplestatus'), nullable=False))
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('regression_dataset', 'status')
    # ### end Alembic commands ###