Skip to content

Commit b1d45bb

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

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

src/lightning/pytorch/callbacks/progress/progress_bar.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -85,11 +85,7 @@ def total_train_batches(self) -> Union[int, float]:
8585
dataloader is of infinite size.
8686
8787
"""
88-
if (
89-
self.trainer.max_epochs == -1
90-
and self.trainer.max_steps is not None
91-
and self.trainer.max_steps > 0
92-
):
88+
if self.trainer.max_epochs == -1 and self.trainer.max_steps is not None and self.trainer.max_steps > 0:
9389
remaining_steps = self.trainer.max_steps - self.trainer.global_step
9490
return min(self.trainer.num_training_batches, remaining_steps)
9591
return self.trainer.num_training_batches

0 commit comments

Comments
 (0)