Skip to content

Commit a52381f

Browse files
committed
fix sanity check breaking on len=0 val dataloader
1 parent 4ff8ff7 commit a52381f

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -390,8 +390,7 @@ def on_sanity_check_start(self, trainer: "pl.Trainer", pl_module: "pl.LightningM
390390

391391
@override
392392
def on_sanity_check_end(self, trainer: "pl.Trainer", pl_module: "pl.LightningModule") -> None:
393-
if self.progress is not None:
394-
assert self.val_sanity_progress_bar_id is not None
393+
if self.progress is not None and self.val_sanity_progress_bar_id is not None:
395394
self.progress.update(self.val_sanity_progress_bar_id, advance=0, visible=False)
396395
self.refresh()
397396

0 commit comments

Comments
 (0)