Skip to content

Commit 2f24cb7

Browse files
committed
add type annotation
1 parent 86d823a commit 2f24cb7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def time_remaining(self) -> Optional[float]:
7070
return None
7171

7272
class _RefreshThread(_RichRefreshThread):
73-
def __init__(self, *args, **kwargs) -> None:
73+
def __init__(self, *args: Any, **kwargs: Any) -> None:
7474
self.refresh_cond = False
7575
super().__init__(*args, **kwargs)
7676

@@ -407,7 +407,7 @@ def _init_progress(self, trainer: "pl.Trainer") -> None:
407407
# progress has started
408408
self._progress_stopped = False
409409

410-
def refresh(self, hard=False) -> None:
410+
def refresh(self, hard: bool = False) -> None:
411411
if self.progress:
412412
if hard or _IS_INTERACTIVE:
413413
self.progress.refresh()

0 commit comments

Comments
 (0)