From dcb24837e10307c1a226e76c549772c20663661a Mon Sep 17 00:00:00 2001 From: Steven Palma Date: Thu, 24 Apr 2025 17:14:56 +0200 Subject: [PATCH] fix(tests): remove verbose argument deprecated in torch 2.7.0 --- tests/optim/test_schedulers.py | 3 --- 1 file changed, 3 deletions(-) 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