Skip to content

Commit edc88bd

Browse files
committed
update
1 parent 340e1d8 commit edc88bd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -448,9 +448,10 @@ def _add_task(self, total_batches: Union[int, float], description: str, visible:
448448
)
449449

450450
def _initialize_progress_bar_id(self) -> None:
451+
if self.is_disabled:
452+
return
451453
total_batches = self.total_train_batches
452454
train_description = self._get_train_description(self.trainer.current_epoch)
453-
assert self.progress is not None
454455
self.train_progress_bar_id = self._add_task(total_batches, train_description)
455456

456457
def _update(self, progress_bar_id: Optional["TaskID"], current: int, visible: bool = True) -> None:

0 commit comments

Comments
 (0)