Coverage for mlos_bench/mlos_bench/storage/sql/alembic/versions/f83fb8ae7fc4_add_trial_runner_id_column.py: 82%

1# 

2# Copyright (c) Microsoft Corporation. 

3# Licensed under the MIT License. 

4# 

5""" 

6Add trial_runner_id column. 

7 

8Revision ID: f83fb8ae7fc4 

9Revises: d2a708351ba8 

10Create Date: 2025-01-03 21:25:48.848196+00:00 

11""" 

12# pylint: disable=no-member 

13 

14from collections.abc import Sequence 

15 

16import sqlalchemy as sa 

17from alembic import op 

18 

19# revision identifiers, used by Alembic. 

20revision: str = "f83fb8ae7fc4" 

21down_revision: str | None = "d2a708351ba8" 

22branch_labels: str | Sequence[str] | None = None 

23depends_on: str | Sequence[str] | None = None 

24 

25 

26def upgrade() -> None: 

27 """The schema upgrade script for this revision.""" 

28 # ### commands auto generated by Alembic - please adjust! ### 

29 op.add_column("trial", sa.Column("trial_runner_id", sa.Integer(), nullable=True, default=None)) 

30 # ### end Alembic commands ### 

31 

32 

33def downgrade() -> None: 

34 """The schema downgrade script for this revision.""" 

35 # ### commands auto generated by Alembic - please adjust! ### 

36 op.drop_column("trial", "trial_runner_id") 

37 # ### end Alembic commands ###