Skip to content

Commit 49bb487

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent b51cabc commit 49bb487

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

src/lightning/pytorch/loggers/mlflow.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -331,10 +331,7 @@ def after_save_checkpoint(self, checkpoint_callback: ModelCheckpoint) -> None:
331331
# log checkpoints as artifacts
332332
if self._log_model == "all" or self._log_model is True and checkpoint_callback.save_top_k == -1:
333333
self._scan_and_log_checkpoints(checkpoint_callback)
334-
elif (
335-
self._log_model is True
336-
and checkpoint_callback not in self._checkpoint_callbacks
337-
):
334+
elif self._log_model is True and checkpoint_callback not in self._checkpoint_callbacks:
338335
self._checkpoint_callbacks.append(checkpoint_callback)
339336

340337
def _scan_and_log_checkpoints(self, checkpoint_callback: ModelCheckpoint) -> None:

0 commit comments

Comments
 (0)