@@ -676,7 +676,7 @@ def formfield(self, form_class=None, choices_form_class=None, **kwargs):
676
676
# we try to pass in. Very annoying because we have to
677
677
# un-encapsulate some of this initialization logic, this makes our
678
678
# EnumChoiceField pretty ugly!
679
- from django_enum .forms import ChoiceFieldMixin , EnumChoiceField , NonStrictSelect
679
+ from django_enum .forms import EnumChoiceField , NonStrictSelect
680
680
681
681
if not self .strict :
682
682
kwargs .setdefault ("widget" , NonStrictSelect )
@@ -687,12 +687,9 @@ def formfield(self, form_class=None, choices_form_class=None, **kwargs):
687
687
** kwargs ,
688
688
)
689
689
690
- if isinstance (form_field , ChoiceFieldMixin ):
691
- # we can't pass these in kwargs because formfield() strips them out
692
- form_field .enum = self .enum
693
- form_field .strict = self .strict
694
- form_field .primitive = self .primitive
695
- # pragma: no cover
690
+ form_field .enum = self .enum
691
+ form_field .strict = self .strict
692
+ form_field .primitive = self .primitive
696
693
return form_field
697
694
698
695
def get_choices (
@@ -1239,7 +1236,6 @@ def contribute_to_class(
1239
1236
1240
1237
def formfield (self , form_class = None , choices_form_class = None , ** kwargs ):
1241
1238
from django_enum .forms import (
1242
- ChoiceFieldMixin ,
1243
1239
EnumFlagField ,
1244
1240
FlagSelectMultiple ,
1245
1241
NonStrictFlagSelectMultiple ,
@@ -1261,11 +1257,9 @@ def formfield(self, form_class=None, choices_form_class=None, **kwargs):
1261
1257
)
1262
1258
1263
1259
# we can't pass these in kwargs because formfield() strips them out
1264
- if isinstance (form_field , ChoiceFieldMixin ):
1265
- form_field .enum = self .enum
1266
- form_field .strict = self .strict
1267
- form_field .primitive = self .primitive
1268
- # pragma: no cover
1260
+ form_field .enum = self .enum # type: ignore
1261
+ form_field .strict = self .strict # type: ignore
1262
+ form_field .primitive = self .primitive # type: ignore
1269
1263
return form_field
1270
1264
1271
1265
def get_choices (
0 commit comments