code: def upgrade():
      with op.batch_alter_table("predictor", schema=None) as batch_op:
          batch_op.add_column(
              sa.Column(
                  "training_data_columns_count",
                  sa.Integer(),
                  nullable=True,
              )
          )
          batch_op.add_column(
              sa.Column(
                  "training_data_rows_count",
                  sa.Integer(),
                  nullable=True,
              )
          )
  
imports: from alembic import op
  import sqlalchemy
  import mindsdb.interfaces.storage.db
label_desc: Use image node_id 1 for calling a global function with example usage: upgrade() without return types