run_stats core fix

materialsproject · Feb 18, 2021 · f08c578 · f08c578
1 parent c17091a
commit f08c578
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/emmet-core/emmet/core/vasp/task.py b/emmet-core/emmet/core/vasp/task.py
@@ -91,7 +91,10 @@ class RunStatistics(BaseModel):
     total_time: float = Field(
         None, description="The total CPU time for this calculation"
     )
-    cores: int = Field(None, description="The number of cores used by VASP")
+    cores: Union[int, str] = Field(
+        None,
+        description="The number of cores used by VASP (some clusters print `mpi-ranks` here)",
+    )
 
 
 class TaskDocument(StructureMetadata):