@@ -672,19 +672,24 @@ def test_default_checkpoint_behavior(tmpdir):
672672@pytest .mark .parametrize ('max_epochs' , [1 , 2 ])
673673@pytest .mark .parametrize ('should_validate' , [True , False ])
674674@pytest .mark .parametrize ('save_last' , [True , False ])
675- def test_model_checkpoint_save_last_warning (tmpdir , caplog , max_epochs , should_validate , save_last ):
675+ @pytest .mark .parametrize ('verbose' , [True , False ])
676+ def test_model_checkpoint_save_last_warning (tmpdir , caplog , max_epochs , should_validate , save_last , verbose ):
676677 """Tests 'Saving latest checkpoint...' log"""
677678 model = LogInTwoMethods ()
678679 if not should_validate :
679680 model .validation_step = None
680681 trainer = Trainer (
681682 default_root_dir = tmpdir ,
682- callbacks = [ModelCheckpoint (monitor = 'early_stop_on' , dirpath = tmpdir , save_top_k = 0 , save_last = save_last )],
683+ callbacks = [
684+ ModelCheckpoint (
685+ monitor = 'early_stop_on' , dirpath = tmpdir , save_top_k = 0 , save_last = save_last , verbose = verbose
686+ )
687+ ],
683688 max_epochs = max_epochs ,
684689 )
685690 with caplog .at_level (logging .INFO ):
686691 trainer .fit (model )
687- assert caplog .messages .count ('Saving latest checkpoint...' ) == save_last
692+ assert caplog .messages .count ('Saving latest checkpoint...' ) == ( verbose and save_last )
688693
689694
690695def test_model_checkpoint_save_last_checkpoint_contents (tmpdir ):
0 commit comments