tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix(tests): remove verbose argument deprecated in torch 2.7.0

Browse Source
This commit is contained in:
Steven Palma
2025-04-24 17:14:56 +02:00
parent c7b204f4a6
commit dcb24837e1
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/optim/test_schedulers.py
View File

@@ -37,7 +37,6 @@ def test_diffuser_scheduler(optimizer):
"base_lrs": [0.001], "base_lrs": [0.001],
"last_epoch": 1, "last_epoch": 1,
"lr_lambdas": [None], "lr_lambdas": [None],
"verbose": False,
} }
assert scheduler.state_dict() == expected_state_dict assert scheduler.state_dict() == expected_state_dict
@@ -56,7 +55,6 @@ def test_vqbet_scheduler(optimizer):
"base_lrs": [0.001], "base_lrs": [0.001],
"last_epoch": 1, "last_epoch": 1,
"lr_lambdas": [None], "lr_lambdas": [None],
"verbose": False,
} }
assert scheduler.state_dict() == expected_state_dict assert scheduler.state_dict() == expected_state_dict
@@ -77,7 +75,6 @@ def test_cosine_decay_with_warmup_scheduler(optimizer):
"base_lrs": [0.001], "base_lrs": [0.001],
"last_epoch": 1, "last_epoch": 1,
"lr_lambdas": [None], "lr_lambdas": [None],
"verbose": False,
} }
assert scheduler.state_dict() == expected_state_dict assert scheduler.state_dict() == expected_state_dict