changed layout of noisy_sgd dict

google-deepmind · Nov 30, 2024 · a162271 · a162271
1 parent 4c1d306
commit a162271
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/optax/_src/alias_test.py b/optax/_src/alias_test.py
@@ -63,7 +63,10 @@
     ),
     dict(opt_name='nadam', opt_kwargs=dict(learning_rate=1e-2)),
     dict(opt_name='nadamw', opt_kwargs=dict(learning_rate=1e-2)),
-    dict(opt_name='noisy_sgd', opt_kwargs=dict(learning_rate=1e-3, key=jrd.key(0), eta=1e-4)),
+    dict(
+        opt_name='noisy_sgd', 
+        opt_kwargs=dict(learning_rate=1e-3, key=jrd.key(0), eta=1e-4)
+    ),
     dict(opt_name='novograd', opt_kwargs=dict(learning_rate=1e-3)),
     dict(
         opt_name='optimistic_gradient_descent',