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