831e2ebde9122f2ffb2d128c4f8b8e85151efad0,torch_geometric/transform/polar_test.py,PolarTest,test_polar_adj,#PolarTest#,11

Before Change

        expected_adj_rho = [
            [0, 1 / sqrt(8), 0],
            [1 / sqrt(8), 0, 1],
            [0, 1, 0],
        ]
        expected_adj_theta = [
            [0, 0.5, 0],

After Change

        weight = torch.FloatTensor([1, 1, 1, 1])
        adj = torch.sparse.FloatTensor(index, weight, torch.Size([3, 3]))
        position = torch.FloatTensor([[1, 0], [0, 0], [-2, 2]])
        data = Data(None, adj, position, None)

        data = PolarAdj()(data)
        adj, position = data.adj, data.position

        expected_adj = [
            [[0, 0], [1 / sqrt(8), 0.5], [0, 0]],
            [[1 / sqrt(8), 0], [0, 0], [1, 0.375]],

Instances