code: def upgrade():
      with op.batch_alter_table("trials") as batch_op:
          batch_op.alter_column(
              "state",
              type_=sa.Enum(
                  "RUNNING",
                  "COMPLETE",
                  "PRUNED",
                  "FAIL",
                  "WAITING",
                  name="trialstate",
              ),
              existing_type=sa.Enum(
                  "RUNNING",
                  "COMPLETE",
                  "PRUNED",
                  "FAIL",
                  name="trialstate",
              ),
          )
  
label_desc: Use image node_id 1 for calling a global function with example usage: upgrade() without return types