import sqlalchemy as sa
from alembic import op

revision = '009_add_annotation_review_fields'
down_revision = '008_add_training_platform_fields'
branch_labels = None
depends_on = None

def upgrade() -> None:
    op.add_column('annotation_tasks', sa.Column('reviewed_at', sa.DateTime(), nullable=True), schema='hub_global')
    op.add_column('annotation_tasks', sa.Column('approved', sa.Boolean(), nullable=True), schema='hub_global')

def downgrade() -> None:
    op.drop_column('annotation_tasks', 'approved', schema='hub_global')
    op.drop_column('annotation_tasks', 'reviewed_at', schema='hub_global')