Skip to content

Commit 1693e96

Browse files
committed
Fix loop reset test
1 parent 7e22306 commit 1693e96

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/tests_pytorch/loops/test_loops.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -601,9 +601,9 @@ def test_fit_loop_reset(tmp_path):
601601
assert epoch_loop.batch_progress.total.ready == 4
602602
assert epoch_loop.batch_progress.total.processed == 4
603603
assert epoch_loop.batch_progress.total.completed == 4 # the checkpoint was saved on train_batch_end
604-
assert epoch_loop.batch_progress.current.ready == 4 # currents get set to the completed value
605-
assert epoch_loop.batch_progress.current.processed == 4
606-
assert epoch_loop.batch_progress.current.completed == 4
604+
assert epoch_loop.batch_progress.current.ready == 0 # currents get set to the completed value
605+
assert epoch_loop.batch_progress.current.processed == 0
606+
assert epoch_loop.batch_progress.current.completed == 0
607607

608608

609609
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)