83025c3d4ad087c1581c8a358333c77e913ca030,torch_geometric/utils/data.py,Data,num_edges,#Data#,28

Before Change

    def num_edges(self):
        if self.edge_index is not None:
            return self.edge_index.size(1)
        elif self.edge_attr is not None:
            return self.edge_attr.size(0)
        else:
            return None

    def cuda(self, *props):

After Change

    @property
    def num_edges(self):
        for _, value in self("edge_attr"):
            return value.size(0)
        return None if self.edge_index is None else self.edge_index.size(1)

Instances