File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
tests/tests_pytorch/loops Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff 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 (
You can’t perform that action at this time.
0 commit comments