py2.7 fix

test/test_basis.py

@@ -26,7 +26,7 @@ def test_spline_basis_cpu(tensor, i):
 
     basis, index = spline_basis_forward(degree, pseudo, kernel_size,
                                         is_open_spline, K)
-    basis = [pytest.approx(x, 0.01) for x in basis.view(-1).tolist()]
+    basis = [pytest.approx(b, 0.01) for b in basis.view(-1).tolist()]
 
     assert basis == expected_basis.view(-1).tolist()
     assert index.tolist() == expected_index.tolist()
@@ -47,7 +47,7 @@ def test_spline_basis_gpu(tensor, i):
     basis, index = spline_basis_forward(degree, pseudo, kernel_size,
                                         is_open_spline, K)
     basis, index = basis.cpu(), index.cpu()
-    basis = [pytest.approx(x, 0.01) for x in basis.view(-1).tolist()]
+    basis = [pytest.approx(b, 0.01) for b in basis.view(-1).tolist()]
 
     assert basis == expected_basis.view(-1).tolist()
     assert index.tolist() == expected_index.tolist()

test/test_spline_conv.py

@@ -38,7 +38,7 @@ def test_spline_conv_cpu(tensor):
         1 + 12.5 * 7 + 13 * 8,
     ]
 
-    output = [pytest.approx(x, 0.01) for x in output.view(-1).tolist()]
+    output = [pytest.approx(o, 0.01) for o in output.view(-1).tolist()]
     assert output == expected_output
 
     x, weight, pseudo = Variable(x), Variable(weight), Variable(pseudo)
@@ -47,7 +47,7 @@ def test_spline_conv_cpu(tensor):
     output = spline_conv(x, edge_index, pseudo, weight, kernel_size,
                          is_open_spline, 1, root_weight, bias)
 
-    output = [pytest.approx(x, 0.01) for x in output.data.view(-1).tolist()]
+    output = [pytest.approx(o, 0.01) for o in output.data.view(-1).tolist()]
     assert output == expected_output