Apply suggestions from code review

Co-authored-by: Adrian Wälchli <[email protected]>
Lightning-AI · Sep 16, 2021 · 0ffc848 · 0ffc848
1 parent bc1dabe
commit 0ffc848
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytorch_lightning/callbacks/progress/rich_progress.py b/pytorch_lightning/callbacks/progress/rich_progress.py
@@ -30,7 +30,7 @@ class CustomTimeColumn(ProgressColumn):
         # Only refresh twice a second to prevent jitter
         max_refresh = 0.5
 
-        def __init__(self, style: Union[str, Style]):
+        def __init__(self, style: Union[str, Style]) -> None:
             self.style = style
             super().__init__()
 
@@ -137,7 +137,7 @@ def __init__(
         self,
         refresh_rate: float = 1.0,
         theme: RichProgressBarTheme = RichProgressBarTheme(),
-    ):
+    ) -> None:
         if not _RICH_AVAILABLE:
             raise ImportError(
                 "`RichProgressBar` requires `rich` to be installed. Install it by running `pip install rich`."