from constants import DATASETS
from flask_wtf import FlaskForm
from wtforms import SelectField, SubmitField


# NOTE The choices should be shorter
class DatasetForm(FlaskForm):
    dataset = SelectField(choices=DATASETS)
    submit = SubmitField("Submit")