Fix lint error

4c16e41b · yanbing-j · d6a017ee · 4c16e41b
Commit 4c16e41b authored Aug 30, 2022 by yanbing-j
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

test/test_conv.py test/test_conv.py +2 -1

No files found.
--- a/test/test_conv.py
+++ b/test/test_conv.py
@@ -56,7 +56,8 @@ def test_spline_conv_forward(test, dtype, device):
                      is_open_spline, 1, True, root_weight, bias)
    if dtype == torch.bfloat16:
        target = torch.tensor(test['expected'])
-        assert torch.allclose(out.to(torch.float), target, rtol=1e-2, atol=1e-2)
+        assert torch.allclose(out.to(torch.float), target,
+                              rtol=1e-2, atol=1e-2)
    else:
        assert out.tolist() == test['expected']