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=invalid-name 

13# pylint: disable=no-member 

14 

15from collections.abc import Sequence 

16 

17import sqlalchemy as sa 

18from alembic import op 

19 

20# revision identifiers, used by Alembic. 

21revision: str = "8928a401115b" 

22down_revision: str | None = "f83fb8ae7fc4" 

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

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

25 

26 

27def upgrade() -> None: 

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

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

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

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

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

33 op.add_column( 

34 "experiment", 

35 sa.Column( 

36 "driver_name", 

37 sa.String(length=40), 

38 nullable=True, 

39 comment="Driver Host/Container Name", 

40 ), 

41 ) 

42 op.add_column( 

43 "experiment", 

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

45 ) 

46 # ### end Alembic commands ### 

47 

48 

49def downgrade() -> None: 

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

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

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

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

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

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

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

57 # ### end Alembic commands ###