"""Add normalization_params to models table

Revision ID: add_normalization_params
Revises: 001_initial_tables
Create Date: 2025-08-12 14:20:00.000000

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = 'add_normalization_params'
down_revision = '001_initial_tables'
branch_labels = None
depends_on = None


def upgrade():
    """Add normalization_params column to models table"""
    op.add_column('models', sa.Column('normalization_params', sa.JSON(), nullable=True))


def downgrade():
    """Remove normalization_params column from models table"""
    op.drop_column('models', 'normalization_params')