Skip to content

Commit 0ffc848

Browse files
Sean Narenawaelchli
andauthored
Apply suggestions from code review
Co-authored-by: Adrian Wälchli <[email protected]>
1 parent bc1dabe commit 0ffc848

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pytorch_lightning/callbacks/progress/rich_progress.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ class CustomTimeColumn(ProgressColumn):
3030
# Only refresh twice a second to prevent jitter
3131
max_refresh = 0.5
3232

33-
def __init__(self, style: Union[str, Style]):
33+
def __init__(self, style: Union[str, Style]) -> None:
3434
self.style = style
3535
super().__init__()
3636

@@ -137,7 +137,7 @@ def __init__(
137137
self,
138138
refresh_rate: float = 1.0,
139139
theme: RichProgressBarTheme = RichProgressBarTheme(),
140-
):
140+
) -> None:
141141
if not _RICH_AVAILABLE:
142142
raise ImportError(
143143
"`RichProgressBar` requires `rich` to be installed. Install it by running `pip install rich`."

0 commit comments

Comments
 (0)