Sanitize Enums

materialsproject · Oct 31, 2020 · 9e0a47f · 9e0a47f
1 parent 27a51a7
commit 9e0a47f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/emmet-core/emmet/core/utils.py b/emmet-core/emmet/core/utils.py
@@ -94,6 +94,8 @@ def jsanitize(obj, strict=False, allow_bson=False):
         return [
             jsanitize(i, strict=strict, allow_bson=allow_bson) for i in obj.tolist()
         ]
+    if isinstance(obj, Enum):
+        return obj.value
     if isinstance(obj, dict):
         return {
             k.__str__(): jsanitize(v, strict=strict, allow_bson=allow_bson)