From 06df293f6c5701890edba1c5d9e21ce06a37d61b Mon Sep 17 00:00:00 2001 From: phi-friday Date: Fri, 29 Apr 2022 22:49:46 +0900 Subject: [PATCH] fix parse_obj, validate method return type fix SQLModel.parse_obj, SQLModel.validate method return type "SQLModel" -> _TSQLModel [issues link](https://github.com/tiangolo/sqlmodel/issues/317) --- sqlmodel/main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/sqlmodel/main.py b/sqlmodel/main.py index 4d6d2f2712..1402305380 100644 --- a/sqlmodel/main.py +++ b/sqlmodel/main.py @@ -564,8 +564,8 @@ def from_orm( @classmethod def parse_obj( - cls: Type["SQLModel"], obj: Any, update: Optional[Dict[str, Any]] = None - ) -> "SQLModel": + cls: Type[_TSQLModel], obj: Any, update: Optional[Dict[str, Any]] = None + ) -> _TSQLModel: obj = cls._enforce_dict_if_root(obj) # SQLModel, support update dict if update is not None: @@ -579,7 +579,7 @@ def __repr_args__(self) -> Sequence[Tuple[Optional[str], Any]]: # From Pydantic, override to enforce validation with dict @classmethod - def validate(cls: Type["SQLModel"], value: Any) -> "SQLModel": + def validate(cls: Type[_TSQLModel], value: Any) -> _TSQLModel: if isinstance(value, cls): return value.copy() if cls.__config__.copy_on_model_validation else value