diff --git a/src/pyplaml/class_attribute.py b/src/pyplaml/class_attribute.py index ca488b2d2b98396a6f5bdb9830e8f4a534b0b215..ea69286a3a0a22a435523940783bd4b594346f5f 100644 --- a/src/pyplaml/class_attribute.py +++ b/src/pyplaml/class_attribute.py @@ -13,7 +13,8 @@ class AttributeModifier(Enum): @staticmethod def from_string(string: str) -> AttributeModifier: - return AttributeModifier.__members__.get(string, AttributeModifier.NONE) + _mod = {member.value: member for member in AttributeModifier}.get(string) + return AttributeModifier.NONE if _mod is None else _mod class ClassAttribute: