Skip to content

Commit 47b07c5

Browse files
committed
update: test case
1 parent d72b81b commit 47b07c5

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/test_utils.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -265,6 +265,9 @@ def test_cpu_offload_optimizer():
265265

266266
def test_orthograd_name():
267267
optimizer = build_orthograd(Example().parameters())
268+
optimizer.zero_grad()
269+
268270
_ = optimizer.param_groups
269-
_ = optimizer.__getstate__()
271+
_ = optimizer.state
272+
270273
assert str(optimizer).lower() == 'orthograd'

0 commit comments

Comments
 (0)