"""Keep the forecaster's in-progress danger ratings on draft runs.

Revision ID: 0002
Revises: 0001
Create Date: 2026-09-14
"""

from alembic import op

revision = "0002"
down_revision = "0001"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.execute("ALTER TABLE tool_runs ADD COLUMN draft_ratings jsonb")


def downgrade() -> None:
    op.execute("ALTER TABLE tool_runs DROP COLUMN draft_ratings")