From 1693e9608c4b4249c51e09b42901b7976cda09b1 Mon Sep 17 00:00:00 2001 From: Luca Antiga Date: Mon, 4 Nov 2024 11:06:36 +0000 Subject: [PATCH] Fix loop reset test --- tests/tests_pytorch/loops/test_loops.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/tests_pytorch/loops/test_loops.py b/tests/tests_pytorch/loops/test_loops.py index 9ec6c6e75e0bd..e1ede580e8324 100644 --- a/tests/tests_pytorch/loops/test_loops.py +++ b/tests/tests_pytorch/loops/test_loops.py @@ -601,9 +601,9 @@ def test_fit_loop_reset(tmp_path): assert epoch_loop.batch_progress.total.ready == 4 assert epoch_loop.batch_progress.total.processed == 4 assert epoch_loop.batch_progress.total.completed == 4 # the checkpoint was saved on train_batch_end - assert epoch_loop.batch_progress.current.ready == 4 # currents get set to the completed value - assert epoch_loop.batch_progress.current.processed == 4 - assert epoch_loop.batch_progress.current.completed == 4 + assert epoch_loop.batch_progress.current.ready == 0 # currents get set to the completed value + assert epoch_loop.batch_progress.current.processed == 0 + assert epoch_loop.batch_progress.current.completed == 0 @pytest.mark.parametrize(