Skip to content

Commit 74ba33d

Browse files
authored
Fix NNTemplateCore callback to always inject the metadata
1 parent f55551b commit 74ba33d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/nn_core/callbacks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def on_train_end(self, trainer: pl.Trainer, pl_module: pl.LightningModule) -> No
5151
def on_save_checkpoint(
5252
self, trainer: pl.Trainer, pl_module: pl.LightningModule, checkpoint: Dict[str, Any]
5353
) -> None:
54-
if self._is_nnlogger(trainer):
54+
if trainer.logger is not None:
5555
trainer.logger.on_save_checkpoint(trainer=trainer, pl_module=pl_module, checkpoint=checkpoint)
5656
metadata = getattr(trainer.datamodule, "metadata", None)
5757
if metadata is not None:

0 commit comments

Comments
 (0)