Fix test error

This commit is contained in:
Daniel Sheppard 2020-02-11 20:27:02 -06:00
parent 3c3cca8ec1
commit 56bb053146

View File

@ -73,7 +73,7 @@ class ScriptVariable:
"""
form_field = self.form_field(**self.field_attrs)
if not isinstance(form_field.widget, forms.CheckboxInput):
if form_field.widget.attrs and form_field.widget.attrs['class']:
if form_field.widget.attrs and 'class' in form_field.widget.attrs.keys():
form_field.widget.attrs['class'] += ' form-control'
else:
form_field.widget.attrs['class'] = 'form-control'