af83e4b143f235d5c41acd80da71c7a20bb8a076,torch_geometric/nn/conv/feast_conv.py,FeaStConv,forward,#FeaStConv#Any#Any#,66

Before Change

        if isinstance(edge_index, Tensor):
            edge_index, _ = remove_self_loops(edge_index)
            num_nodes = x[1].size(0)
            edge_index, _ = add_self_loops(edge_index, num_nodes=num_nodes)
        elif isinstance(edge_index, SparseTensor):
            edge_index = set_diag(edge_index)

        // propagate_type: (x: PairTensor)

After Change

        if self.add_self_loops:
            if isinstance(edge_index, Tensor):
                edge_index, _ = remove_self_loops(edge_index)
                edge_index, _ = add_self_loops(edge_index,
                                               num_nodes=x[1].size(0))
            elif isinstance(edge_index, SparseTensor):
                edge_index = set_diag(edge_index)

        // propagate_type: (x: PairTensor)

Instances