The code below worked for my situation

def upgrade():
    companytype = sa.Column('type', sa.Enum('type1', 'type2', name='companytype')
    companytype.create(op.get_bind())

    op.add_column('company', companytype, nullable=True))