Skip to content

Commit

Permalink
switch vasp task_doc to new base
Browse files Browse the repository at this point in the history
  • Loading branch information
shyamd committed May 13, 2021
1 parent 88fcc26 commit cb97771
Showing 1 changed file with 3 additions and 10 deletions.
13 changes: 3 additions & 10 deletions emmet-core/emmet/core/vasp/task.py
Expand Up @@ -13,6 +13,7 @@
from emmet.core.math import Matrix3D, Vector3D
from emmet.core.mpid import MPID
from emmet.core.structure import StructureMetadata
from emmet.core.task import TaskDocument as BaseTaskDocument
from emmet.core.utils import ValueEnum
from emmet.core.vasp.calc_types import (
CalcType,
Expand Down Expand Up @@ -98,22 +99,16 @@ class RunStatistics(BaseModel):
)


class TaskDocument(StructureMetadata):
class TaskDocument(BaseTaskDocument, StructureMetadata):
"""
Definition of VASP Task Document
"""

dir_name: str = Field(None, description="The directory for this VASP task")
calc_code: ClassVar[str] = "VASP"
run_stats: Dict[str, RunStatistics] = Field(
{},
description="Summary of runtime statisitics for each calcualtion in this task",
)
completed_at: datetime = Field(
None, description="Timestamp for when this task was completed"
)
last_updated: datetime = Field(
None, description="Timestamp for this task document was last updateed"
)

is_valid: bool = Field(
True, description="Whether this task document passed validation or not"
Expand All @@ -127,8 +122,6 @@ class TaskDocument(StructureMetadata):
orig_inputs: Dict[str, Any] = Field(
{}, description="Summary of the original VASP inputs"
)
task_id: MPID = Field(None, description="the Task ID For this document")
tags: List[str] = Field([], description="Metadata tags for this task document")

calcs_reversed: List[Dict] = Field(
[], description="The 'raw' calculation docs used to assembled this task"
Expand Down

0 comments on commit cb97771

Please sign in to comment.