ensure value enums can be compared with strings

materialsproject · Jun 29, 2021 · 4a188af · 4a188af
1 parent 878a91f
commit 4a188af
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/emmet-core/emmet/core/utils.py b/emmet-core/emmet/core/utils.py
@@ -139,6 +139,12 @@ class ValueEnum(Enum):
     def __str__(self):
         return str(self.value)
 
+    def __eq__(self, o: object) -> bool:
+        """Special Equals to enable converting strings back to the enum"""
+        if isinstance(o, str):
+            return super().__eq__(self.__class__(o))
+        return super().__eq__(o)
+
 
 class DocEnum(ValueEnum):
     """