def upgrade(): op.create_primary_key('pfizer_pkey', 'pfizer', ['nct_id']) op.create_primary_key('takeda_pkey', 'takeda', ['takeda_trial_id'])