Skip to content

Commit 9b10555

Browse files
fix: deprecation warning wihen using sqlmodel_update
1 parent 7256820 commit 9b10555

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

sqlmodel/main.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -992,17 +992,17 @@ def sqlmodel_update(
992992
use_update = (update or {}).copy()
993993
if isinstance(obj, dict):
994994
for key, value in {**obj, **use_update}.items():
995-
if key in get_model_fields(self):
995+
if key in get_model_fields(self.__class__):
996996
setattr(self, key, value)
997997
elif isinstance(obj, BaseModel):
998-
for key in get_model_fields(obj):
998+
for key in get_model_fields(obj.__class__):
999999
if key in use_update:
10001000
value = use_update.pop(key)
10011001
else:
10021002
value = getattr(obj, key)
10031003
setattr(self, key, value)
10041004
for remaining_key in use_update:
1005-
if remaining_key in get_model_fields(self):
1005+
if remaining_key in get_model_fields(self.__class__):
10061006
value = use_update.pop(remaining_key)
10071007
setattr(self, remaining_key, value)
10081008
else:

0 commit comments

Comments
 (0)