ensure enums saved as values

materialsproject · Oct 22, 2020 · 8de0c00 · 8de0c00
1 parent ad3b8d6
commit 8de0c00
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 0 deletions.
diff --git a/emmet-core/emmet/core/provenance.py b/emmet-core/emmet/core/provenance.py
@@ -76,6 +76,9 @@ class Provenance(PropertyDoc):
         description="List of history nodes specifying the transformations or orignation of this material",
     )
 
+    class Config:
+        use_enum_values = True
+
     @validator("authors")
     def remove_duplicate_authors(cls, authors):
         authors_dict = {entry.name.lower(): entry for entry in authors}

diff --git a/emmet-core/emmet/core/symmetry.py b/emmet-core/emmet/core/symmetry.py
@@ -55,6 +55,9 @@ class SymmetryData(BaseModel):
 
     version: str = Field(None, title="SPGLib version")
 
+    class Config:
+        use_enum_values = True
+
     @classmethod
     def from_structure(cls, structure: Structure) -> "SymmetryData":
         symprec = SETTINGS.SYMPREC

diff --git a/emmet-core/emmet/core/vasp/task.py b/emmet-core/emmet/core/vasp/task.py
@@ -124,6 +124,9 @@ class TaskDocument(StructureMetadata):
         None, description="List of sandboxes this task document is allowed in"
     )
 
+    class Config:
+        use_enum_values = True
+
     @property
     def run_type(self) -> RunType:
         return run_type(self.input.parameters)

diff --git a/emmet-core/emmet/core/vasp/validation.py b/emmet-core/emmet/core/vasp/validation.py
@@ -36,6 +36,7 @@ class ValidationDoc(BaseModel):
 
     class Config:
         extra = "allow"
+        use_enum_values = True
 
     @classmethod
     def from_task_doc(

diff --git a/emmet-core/emmet/core/xas.py b/emmet-core/emmet/core/xas.py
@@ -29,6 +29,9 @@ class XASDoc(SpectrumDoc):
         ..., title="Absorption Edge", description="The interaction edge for XAS"
     )
 
+    class Config:
+        use_enum_values = True
+
     @classmethod
     def from_spectrum(
         cls,