diff --git a/digits/dataset/images/forms.py b/digits/dataset/images/forms.py index 0ecb9c1bd..aef0c008a 100644 --- a/digits/dataset/images/forms.py +++ b/digits/dataset/images/forms.py @@ -37,5 +37,5 @@ class ImageDatasetForm(DatasetForm): ) resize_mode = wtforms.SelectField(u'Resize Transformation', default='squash', - choices=ImageDatasetJob.choices() + choices=ImageDatasetJob.resize_mode_choices() ) diff --git a/digits/dataset/images/job.py b/digits/dataset/images/job.py index b4ab146f5..7e86f9c2e 100644 --- a/digits/dataset/images/job.py +++ b/digits/dataset/images/job.py @@ -23,7 +23,7 @@ def __init__(self, image_dims, resize_mode, **kwargs): self.resize_mode = resize_mode @staticmethod - def choices(): + def resize_mode_choices(): return [ ('crop', 'Crop'), ('squash', 'Squash'), @@ -32,6 +32,6 @@ def choices(): ] def resize_mode_name(self): - c = dict(self.choices()) + c = dict(self.resize_mode_choices()) return c[self.resize_mode]