File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change 2424from pytorch_lightning .utilities .exceptions import MisconfigurationException
2525from tests .helpers import BoringModel
2626from tests .helpers .runif import RunIf
27+ from tests .helpers .utils import no_warning_call
2728
2829
2930def test_trainer_flag (caplog ):
@@ -106,7 +107,9 @@ def test_timer_stops_training(tmpdir, caplog):
106107 timer = Timer (duration = duration )
107108
108109 trainer = Trainer (default_root_dir = tmpdir , max_epochs = 1000 , callbacks = [timer ])
109- with caplog .at_level (logging .INFO ):
110+ with caplog .at_level (logging .INFO ), no_warning_call (
111+ DeprecationWarning , match = "The signature of `Callback.on_train_epoch_end` has changed in v1.3"
112+ ):
110113 trainer .fit (model )
111114 assert trainer .global_step > 1
112115 assert trainer .current_epoch < 999
You can’t perform that action at this time.
0 commit comments