From 8e0879cbb168bdfe57711471ada215b9e1ae6189 Mon Sep 17 00:00:00 2001
From: Tyler Mathis <35553152+tsmathis@users.noreply.github.com>
Date: Wed, 25 Sep 2024 17:50:26 -0700
Subject: [PATCH] bring run_stats in as root level field

---
 emmet-core/emmet/core/tasks.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/emmet-core/emmet/core/tasks.py b/emmet-core/emmet/core/tasks.py
index dba9044c2..2d994dd58 100644
--- 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