From c27bdcfb029772b5675c6711e4aa4ad711671281 Mon Sep 17 00:00:00 2001 From: SeanNaren Date: Tue, 2 Nov 2021 10:04:43 +0000 Subject: [PATCH] Only enable pulse for infinite datasets --- pytorch_lightning/callbacks/progress/rich_progress.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/callbacks/progress/rich_progress.py b/pytorch_lightning/callbacks/progress/rich_progress.py index b042adf89788b..d684f8a7e38ed 100644 --- a/pytorch_lightning/callbacks/progress/rich_progress.py +++ b/pytorch_lightning/callbacks/progress/rich_progress.py @@ -37,7 +37,7 @@ def render(self, task: "Task") -> ProgressBar: total=max(0, task.total), completed=max(0, task.completed), width=None if self.bar_width is None else max(1, self.bar_width), - pulse=not task.started or math.isfinite(task.remaining), + pulse=not task.started or not math.isfinite(task.remaining), animation_time=task.get_time(), style=self.style, complete_style=self.complete_style,