"""Use 64-bit scheduler and worker generations.

Revision ID: 20260826_0015
Revises: 20260826_0014
"""

from collections.abc import Sequence

import sqlalchemy as sa

from alembic import op

revision: str = "20260826_0015"
down_revision: str | None = "20260826_0014"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None


def upgrade() -> None:
    op.alter_column(
        "residency_allocations",
        "generation",
        existing_type=sa.Integer(),
        type_=sa.BigInteger(),
        existing_nullable=False,
        existing_server_default="1",
    )
    op.alter_column(
        "serving_gpu_leases",
        "generation",
        existing_type=sa.Integer(),
        type_=sa.BigInteger(),
        existing_nullable=False,
        existing_server_default="1",
    )


def downgrade() -> None:
    op.alter_column(
        "serving_gpu_leases",
        "generation",
        existing_type=sa.BigInteger(),
        type_=sa.Integer(),
        existing_nullable=False,
        existing_server_default="1",
    )
    op.alter_column(
        "residency_allocations",
        "generation",
        existing_type=sa.BigInteger(),
        type_=sa.Integer(),
        existing_nullable=False,
        existing_server_default="1",
    )