Coverage for mindsdb / migrations / versions / 2023-02-25_3154382dab17_training_progress.py: 50%

1"""training_progress 

2 

3Revision ID: 3154382dab17 

4Revises: ee63d868fa84 

5Create Date: 2023-02-25 15:12:02.828938 

6 

7""" 

8from alembic import op 

9import sqlalchemy as sa 

10 

11 

12# revision identifiers, used by Alembic. 

13revision = '3154382dab17' 

14down_revision = 'ee63d868fa84' 

15branch_labels = None 

16depends_on = None 

17 

18 

19def upgrade(): 

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

21 with op.batch_alter_table('predictor', schema=None) as batch_op: 

22 batch_op.add_column(sa.Column('training_phase_current', sa.Integer(), nullable=True)) 

23 batch_op.add_column(sa.Column('training_phase_total', sa.Integer(), nullable=True)) 

24 batch_op.add_column(sa.Column('training_phase_name', sa.String(), nullable=True)) 

25 

26 # ### end Alembic commands ### 

27 

28 

29def downgrade(): 

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

31 with op.batch_alter_table('predictor', schema=None) as batch_op: 

32 batch_op.drop_column('training_phase_name') 

33 batch_op.drop_column('training_phase_total') 

34 batch_op.drop_column('training_phase_current') 

35 

36 # ### end Alembic commands ###