Skip to content

Commit 4e35c22

Browse files
committed
add parent
1 parent cd30ce4 commit 4e35c22

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/pytorch/trainer/connectors/callback_connector.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -240,7 +240,7 @@ def _reorder_callbacks(callbacks: list[Callback]) -> list[Callback]:
240240

241241

242242
def _validate_callbacks_list(callbacks: list[Callback]) -> None:
243-
stateful_callbacks = [cb for cb in callbacks if is_overridden("state_dict", instance=cb)]
243+
stateful_callbacks = [cb for cb in callbacks if is_overridden("state_dict", instance=cb, parent=Callback)]
244244
seen_callbacks = set()
245245
for callback in stateful_callbacks:
246246
if callback.state_key in seen_callbacks:

0 commit comments

Comments
 (0)