diff --git a/tests/test_grad.py b/tests/test_grad.py index 9156d14..e8257ea 100644 --- a/tests/test_grad.py +++ b/tests/test_grad.py @@ -58,9 +58,6 @@ def test_low_order_cpu( x, zi, at, rt = (lambda x: tuple(xx.to(device) for xx in x))( create_test_inputs(batch_size, samples) ) - x, zi, at, rt = (lambda x: tuple(xx.to(device) for xx in x))( - create_test_inputs(batch_size, samples) - ) x.requires_grad = x_requires_grad zi.requires_grad = zi_requires_grad at.requires_grad = at_requires_grad