Coverage for mindsdb / migrations / versions / 2022-07-15_b5b53e0ea7f8_training_data_rows_columns_count.py: 60%

1"""training_data_rows_columns_count 

2 

3Revision ID: b5b53e0ea7f8 

4Revises: 999bceb904df 

5Create Date: 2022-07-15 12:21:40.523039 

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 = 'b5b53e0ea7f8' 

15down_revision = '999bceb904df' 

16branch_labels = None 

17depends_on = None 

18 

19 

20def upgrade(): 

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

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

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

24 

25 

26def downgrade(): 

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

28 batch_op.drop_column('training_data_rows_count') 

29 batch_op.drop_column('training_data_columns_count')