update: test case

kozistr · Jan 21, 2025 · 47b07c5 · 47b07c5
1 parent d72b81b
commit 47b07c5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -265,6 +265,9 @@ def test_cpu_offload_optimizer():
 
 def test_orthograd_name():
     optimizer = build_orthograd(Example().parameters())
+    optimizer.zero_grad()
+
     _ = optimizer.param_groups
-    _ = optimizer.__getstate__()
+    _ = optimizer.state
+
     assert str(optimizer).lower() == 'orthograd'