Does pandera support validating enum.Enum or subclases of it ?

[davidandreoletti]

Assuming a pedantic like class declaration with:

class SizeEnum(enum.Enum):
     BIG = "big"
     SMALL = "small"

class SummaryDFSchema(pandera...):
     size : pandera.Series[SizeEnum]
     name : ...

Currently pandera fails (via exception raised) because it seems pandas does not reconignize the Enum as a registered custom dtype.

What methods/workaround could be used to let pandera enforce/check the column contains SizeEnum types (rather than one of its string values such as "big")?

[cosmicBboy]

hi @davidandreoletti, this is currently not supported, but this would be a great feature! Going to turn this into an enhancement issue.

[cosmicBboy]

created #911 to track this issue.