From 48a7c8e6e9b240d6e5b532288b0899124a2729bc Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sat, 7 Dec 2024 04:43:19 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- tests/tests_pytorch/trainer/optimization/test_optimizers.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/tests_pytorch/trainer/optimization/test_optimizers.py b/tests/tests_pytorch/trainer/optimization/test_optimizers.py index d75d5230bde40..a1b555d31b6a5 100644 --- a/tests/tests_pytorch/trainer/optimization/test_optimizers.py +++ b/tests/tests_pytorch/trainer/optimization/test_optimizers.py @@ -657,6 +657,7 @@ def lr_scheduler_step(*_): ... with pytest.raises(MisconfigurationException, match="CustomScheduler` doesn't follow"): _init_optimizers_and_lr_schedulers(model) + def test_lr_scheduler_step_across_epoch_boundaries(mocked_sched, tmp_path): class StepAcrossEpochsModel(LightningModule): def __init__(self):