Skip to content

Commit d2d9289

Browse files
committed
🎨 Update type annotations in _compat.py
1 parent 15765d4 commit d2d9289

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

sqlmodel/_compat.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
TYPE_CHECKING,
77
AbstractSet,
88
Any,
9+
Callable,
910
Dict,
1011
ForwardRef,
1112
Generator,
@@ -100,7 +101,9 @@ def set_config_value(
100101
def get_model_fields(model: InstanceOrType[BaseModel]) -> Dict[str, "FieldInfo"]:
101102
return model.model_fields
102103

103-
def get_fields_set(object: InstanceOrType["SQLModel"]) -> Union[Set[str], property]:
104+
def get_fields_set(
105+
object: InstanceOrType["SQLModel"]
106+
) -> Union[Set[str], Callable[[BaseModel], set[str]]]:
104107
return object.model_fields_set
105108

106109
def set_fields_set(
@@ -393,7 +396,9 @@ def set_config_value(
393396
def get_model_fields(model: InstanceOrType[BaseModel]) -> Dict[str, "FieldInfo"]:
394397
return model.__fields__ # type: ignore
395398

396-
def get_fields_set(object: InstanceOrType["SQLModel"]) -> Union[Set[str], property]:
399+
def get_fields_set(
400+
object: InstanceOrType["SQLModel"]
401+
) -> Union[Set[str], Callable[[BaseModel], set[str]]]:
397402
return object.__fields_set__
398403

399404
def set_fields_set(

0 commit comments

Comments
 (0)