Coverage for mindsdb / migrations / versions / 2022-07-22_6e834843e7e9_training_time.py: 60%

1"""training_time 

2 

3Revision ID: 6e834843e7e9 

4Revises: b5b53e0ea7f8 

5Create Date: 2022-07-22 13:32:34.796604 

6 

7""" 

8from alembic import op 

9import sqlalchemy as sa 

10import mindsdb.interfaces.storage.db # noqa 

11 

12 

13# revision identifiers, used by Alembic. 

14revision = '6e834843e7e9' 

15down_revision = 'b5b53e0ea7f8' 

16branch_labels = None 

17depends_on = None 

18 

19 

20def upgrade(): 

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

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

23 batch_op.add_column(sa.Column('training_start_at', sa.DateTime(), nullable=True)) 

24 batch_op.add_column(sa.Column('training_stop_at', sa.DateTime(), nullable=True)) 

25 # ### end Alembic commands ### 

26 

27 

28def downgrade(): 

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

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

31 batch_op.drop_column('training_stop_at') 

32 batch_op.drop_column('training_start_at') 

33 # ### end Alembic commands ###