e850d36cab88360177f85d98d73c5efb4d9925ba,torch_geometric/transforms/graclus.py,Graclus,__call__,#Graclus#Any#Any#,10

Before Change

        data.position = positions[0]
        data.input = perm_input(data.input, perm)

        for i in range(2, self.level + 2):
            data.add("adj_{}".format(i), adjs[i - 1])
            data.add("position_{}".format(i), positions[i - 1])

        return data


def adj_distance(adj, position):

After Change

        data.input = perm_input(data.input, perm)
        data.adj = None
        data.add("adjs", adjs)
        data.position = None
        data.add("positions", positions)

        return data

Instances