update: test_no_gradients

kozistr · Jan 21, 2025 · 4a1b3b7 · 4a1b3b7
1 parent 26c3288
commit 4a1b3b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_gradients.py b/tests/test_gradients.py
@@ -37,7 +37,7 @@ def test_no_gradients(optimizer_name):
     sphere_loss(p1 + p3).backward(create_graph=True)
 
     optimizer.step(lambda: 0.1)  # for AliG optimizer
-    if optimizer_name not in {'lookahead', 'trac'}:
+    if optimizer_name not in {'lookahead', 'trac', 'orthograd'}:
         optimizer.zero_grad(set_to_none=True)