From 6dc7f1829f059c78a0606bfcc06c3bd11f7280f2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tom=C3=A1=C5=A1=20Orvisk=C3=BD?= <orvisky.tomas@centrum.cz> Date: Sun, 28 Jan 2024 12:17:40 +0100 Subject: [PATCH] Revert from_string for class attribute --- src/pyplaml/class_attribute.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/pyplaml/class_attribute.py b/src/pyplaml/class_attribute.py index ca488b2..ea69286 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: -- GitLab