def __init__(self) -> None:
super().__init__()
# since manual optimization does not track lr scheduler or optimizer frequencies, we use a simpler progress than
# `OptimizationProgress`
self.optim_step_progress = Progress.from_defaults(ReadyCompletedTracker)
self._done: bool = False
self._hiddens: Optional[Any] = None
self._output: _OUTPUTS_TYPE = {}
def test_progress_from_defaults():
actual = Progress.from_defaults(StartedTracker, completed=5)
expected = Progress(total=StartedTracker(completed=5),
current=StartedTracker(completed=5))
assert actual == expected
def test_base_progress_from_defaults():
actual = Progress.from_defaults(completed=5, started=None)
expected = Progress(total=Tracker(started=None, completed=5), current=Tracker(started=None, completed=5))
assert actual == expected