55 lines
1.3 KiB
Python
55 lines
1.3 KiB
Python
"""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",
|
|
)
|