diff --git a/elementary/utils/pydantic_shim.py b/elementary/utils/pydantic_shim.py index 09f9e371b..4ab24a417 100644 --- a/elementary/utils/pydantic_shim.py +++ b/elementary/utils/pydantic_shim.py @@ -11,8 +11,10 @@ if pydantic_major == "1": from pydantic import ( # type: ignore # noqa BaseModel, + EmailStr, Extra, Field, + ValidationError, create_model, root_validator, validator, @@ -20,8 +22,10 @@ elif pydantic_major == "2": from pydantic.v1 import ( # type: ignore # noqa BaseModel, + EmailStr, Extra, Field, + ValidationError, create_model, root_validator, validator,