c49bd59c7d43af4a3baadbfe17d6d77a6a6040fd,torch_geometric/datasets/particle.py,TrackingData,__inc__,#TrackingData#Any#Any#,12

Before Change

class TrackingData(Data):
    def __inc__(self, key, item):
        if key == "y":
            return item.max().item() + 1
        else:
            return super(TrackingData, self).__inc__(key, item)

After Change

class TrackingData(Data):
    def __inc__(self, key, item):
        if key == "y_index":
            return torch.tensor([item[0].max().item() + 1, self.num_nodes])
        else:
            return super(TrackingData, self).__inc__(key, item)

Instances