bring run_stats in as root level field

materialsproject · Sep 26, 2024 · 02838e4 · 02838e4
1 parent 2b80091
commit 02838e4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/emmet-core/emmet/core/tasks.py b/emmet-core/emmet/core/tasks.py
@@ -434,6 +434,11 @@ class TaskDoc(StructureMetadata, extra="allow"):
         description="Identifier for this calculation; should provide rough information about the calculation origin and purpose.",
     )
 
+    run_stats: Optional[RunStatistics] = Field(
+        None,
+        description="Summary of runtime statistics for each calculation in this task",
+    )
+
     # Note that private fields are needed because TaskDoc permits extra info
     # added to the model, unlike TaskDocument. Because of this, when pydantic looks up
     # attrs on the model, it searches for them in the model extra dict first, and if it