7c41e2eee4ff49865ea9fc2fc94fe91a68cb4807,torch_geometric/datasets/icews.py,TemporalDataset,num_rels,#TemporalDataset#,35

Before Change

    @property
    def num_rels(self):
        if hasattr(self, "_num_rels"):
            return self._num_rels
        ps = self.processed_paths
        rs = [torch.load(path)[0].edge_attr.max().item() + 1 for path in ps]
        self._num_rels = max(rs)
        return self._num_rels

    @property
    def raw_file_names(self):
        return ["{}.txt".format(name) for name in ["train", "valid", "test"]]

After Change

    @property
    def num_rels(self):
        return self.data.edge_type.max().item() + 1

    @property
    def raw_file_names(self):
        return ["{}.txt".format(name) for name in ["train", "valid", "test"]]

Instances