[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
2025-03-24 13:16:38 +00:00
parent cdcf346061
commit 1c8daf11fd
95 changed files with 1592 additions and 491 deletions
--- a/tests/optim/test_schedulers.py
+++ b/tests/optim/test_schedulers.py
@@ -43,7 +43,9 @@ def test_diffuser_scheduler(optimizer):


 def test_vqbet_scheduler(optimizer):
-    config = VQBeTSchedulerConfig(num_warmup_steps=10, num_vqvae_training_steps=20, num_cycles=0.5)
+    config = VQBeTSchedulerConfig(
+        num_warmup_steps=10, num_vqvae_training_steps=20, num_cycles=0.5
+    )
    scheduler = config.build(optimizer, num_training_steps=100)
    assert isinstance(scheduler, LambdaLR)