diff --git a/tests/test_ops/test_conv_gradfix.py b/tests/test_ops/test_conv_gradfix.py index 4a5b50184c..ff2f35c55a 100644 --- a/tests/test_ops/test_conv_gradfix.py +++ b/tests/test_ops/test_conv_gradfix.py @@ -37,6 +37,7 @@ def test_conv2d_transposed_cuda(self): weight = self.weight.cuda() res = conv_transpose2d(x, weight, None, 1, 1) assert res.shape == (1, 1, 32, 32) - gradcheck(conv_transpose2d, (x, weight, None, 1, 1), eps=1e-2, atol=1e-2) + gradcheck( + conv_transpose2d, (x, weight, None, 1, 1), eps=1e-2, atol=1e-2) gradgradcheck( conv_transpose2d, (x, weight, None, 1, 1), eps=1e-2, atol=1e-2)