add computed entry and remove comment

materialsproject · Oct 1, 2020 · 99459db · 99459db
1 parent 83591d0
commit 99459db
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/emmet-core/emmet/core/vasp/task.py b/emmet-core/emmet/core/vasp/task.py
@@ -8,10 +8,7 @@
 from pymatgen.analysis.magnetism import CollinearMagneticStructureAnalyzer, Ordering
 
 from emmet.core.structure import StructureMetadata
-from emmet.stubs import Matrix3D, Structure
-
-
-### Computed Entry for each calculation
+from emmet.stubs import Matrix3D, Structure, ComputedEntry
 
 
 class Status(Enum):
@@ -131,3 +128,6 @@ class TaskDocument(StructureMetadata):
         None, description="List of VASP objects included with this Task Document"
     )
     vasp_objects: Dict[VASPObjects, Any]
+    entry: ComputedEntry = Field(
+        None, description="The computed Entry for this calculation"
+    )