Fix formatting

Lightning-AI · May 14, 2022 · ddf3a3c · ddf3a3c
1 parent 7388e70
commit ddf3a3c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pytorch_lightning/trainer/connectors/checkpoint_connector.py b/pytorch_lightning/trainer/connectors/checkpoint_connector.py
@@ -360,7 +360,7 @@ def dump_checkpoint(self, weights_only: bool = False) -> dict:
  'global_step': training global step
  'pytorch-lightning_version': The version of PyTorch Lightning that produced this checkpoint
  'callbacks': "callback specific state"[] # if not weights_only
- 'loggers': "logger specific state"[] # if not weights_only
+ 'loggers': "logger specific state"[]  # if not weights_only
  'optimizer_states': "PT optim's state_dict"[] # if not weights_only
  'lr_schedulers': "PT sched's state_dict"[] # if not weights_only
  'state_dict': Model's state_dict (e.g. network weights)

diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -1713,6 +1713,7 @@ def _call_loggers_state_dict(self) -> Dict[str, dict]:
  state_dict = logger.state_dict()
  if state_dict:
  logger_state_dicts[logger.state_key] = state_dict
+
  return logger_state_dicts
 
  def _call_loggers_load_state_dict(self, checkpoint: Dict[str, Any]) -> None: