diff --git a/tests/optim/test_schedulers.py b/tests/optim/test_schedulers.py index 17637663..d6191dce 100644 --- a/tests/optim/test_schedulers.py +++ b/tests/optim/test_schedulers.py @@ -37,7 +37,6 @@ def test_diffuser_scheduler(optimizer): "base_lrs": [0.001], "last_epoch": 1, "lr_lambdas": [None], - "verbose": False, } assert scheduler.state_dict() == expected_state_dict @@ -56,7 +55,6 @@ def test_vqbet_scheduler(optimizer): "base_lrs": [0.001], "last_epoch": 1, "lr_lambdas": [None], - "verbose": False, } assert scheduler.state_dict() == expected_state_dict @@ -77,7 +75,6 @@ def test_cosine_decay_with_warmup_scheduler(optimizer): "base_lrs": [0.001], "last_epoch": 1, "lr_lambdas": [None], - "verbose": False, } assert scheduler.state_dict() == expected_state_dict