Skip to content

Commit b51cabc

Browse files
committed
fixed the tests
1 parent 405958d commit b51cabc

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/lightning/pytorch/loggers/mlflow.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ def __init__(
142142
self.tags = tags
143143
self._log_model = log_model
144144
self._logged_model_time: dict[str, float] = {}
145-
self._checkpoint_callbacks: Optional[list[ModelCheckpoint]] = []
145+
self._checkpoint_callbacks: list[ModelCheckpoint] = []
146146
self._prefix = prefix
147147
self._artifact_location = artifact_location
148148
self._log_batch_kwargs = {} if synchronous is None else {"synchronous": synchronous}
@@ -333,7 +333,6 @@ def after_save_checkpoint(self, checkpoint_callback: ModelCheckpoint) -> None:
333333
self._scan_and_log_checkpoints(checkpoint_callback)
334334
elif (
335335
self._log_model is True
336-
and self._checkpoint_callbacks
337336
and checkpoint_callback not in self._checkpoint_callbacks
338337
):
339338
self._checkpoint_callbacks.append(checkpoint_callback)

0 commit comments

Comments
 (0)