Skip to content

Commit 7649795

Browse files
committed
fix: test_soap_parameters
1 parent 034b65c commit 7649795

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_optimizers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -481,7 +481,7 @@ def test_soap_parameters(params):
481481
model[0].weight.grad = torch.zeros((8, 2))
482482
model[0].bias.grad = torch.zeros((8,))
483483
model[1].weight.grad = torch.zeros((1, 8))
484-
model[1].bias.grad = torch.zeros((1,))
484+
model[1].bias.grad = None
485485

486486
optimizer.step()
487487

0 commit comments

Comments
 (0)