"""Add column to pii entities for confidence score

Revision ID: c48da0268fd9
Revises: 2cdcefae64ef
Create Date: 2023-11-13 21:59:06.386621

"""

import sqlalchemy as sa
from alembic import op

# revision identifiers, used by Alembic.
revision = "c48da0268fd9"
down_revision = "2cdcefae64ef"
branch_labels = None
depends_on = None


def upgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column("pii_entities", sa.Column("confidence", sa.Float(), nullable=True))
    # ### end Alembic commands ###


def downgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column("pii_entities", "confidence")
    # ### end Alembic commands ###
