Coverage for mlos_bench/mlos_bench/storage/sql/alembic/versions/8928a401115b_adding_experiment_table_columns_to_.py: 47%

1# 

2# Copyright (c) Microsoft Corporation. 

3# Licensed under the MIT License. 

4# 

5"""Adding Experiment table columns to support mlos_benchd service - See #732 

6 

7Revision ID: 8928a401115b 

8Revises: f83fb8ae7fc4 

9Create Date: 2025-01-14 17:06:36.181503+00:00 

10 

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 = "8928a401115b" 

21down_revision: str | None = "f83fb8ae7fc4" 

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("experiment", sa.Column("ts_start", sa.DateTime(), nullable=True)) 

30 op.add_column("experiment", sa.Column("ts_end", sa.DateTime(), nullable=True)) 

31 op.add_column("experiment", sa.Column("status", sa.String(length=16), nullable=True)) 

32 op.add_column( 

33 "experiment", 

34 sa.Column( 

35 "driver_name", 

36 sa.String(length=40), 

37 nullable=True, 

38 comment="Driver Host/Container Name", 

39 ), 

40 ) 

41 op.add_column( 

42 "experiment", 

43 sa.Column("driver_pid", sa.Integer(), nullable=True, comment="Driver Process ID"), 

44 ) 

45 # ### end Alembic commands ### 

46 

47 

48def downgrade() -> None: 

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

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

51 op.drop_column("experiment", "driver_pid") 

52 op.drop_column("experiment", "driver_name") 

53 op.drop_column("experiment", "status") 

54 op.drop_column("experiment", "ts_end") 

55 op.drop_column("experiment", "ts_start") 

56 # ### end Alembic commands ###