From e5fd206fc7dfa057dc9d280f2291e27553c06a11 Mon Sep 17 00:00:00 2001 From: Jason Munro Date: Mon, 3 Oct 2022 15:34:08 -0700 Subject: [PATCH] Enable potcar single validation in task doc (#552) --- emmet-core/emmet/core/tasks.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/emmet-core/emmet/core/tasks.py b/emmet-core/emmet/core/tasks.py index e742ef543f..d55080dba0 100644 --- a/emmet-core/emmet/core/tasks.py +++ b/emmet-core/emmet/core/tasks.py @@ -1,6 +1,6 @@ from datetime import datetime from enum import Enum -from typing import List, Dict, Union +from typing import List, Dict, Union, Any from emmet.core.vasp.task_valid import TaskState @@ -56,10 +56,18 @@ class OrigInputs(BaseModel): None, description="Pymatgen object representing the KPOINTS file.", ) - potcar: Union[Potcar, List[VaspPotcarSingle], VaspPotcar] = Field( + potcar: Union[Potcar, VaspPotcar, List[Any]] = Field( None, description="Pymatgen object representing the POTCAR file.", ) + # Make sure that the datetime field is properly formatted + @validator("potcar", pre=True) + def potcar_ok(cls, v): + if isinstance(v, list): + return [i for i in v] + + return v + class Config: arbitrary_types_allowed = True