91abf0d795b4876e51ad23f4543f7fc6383fc718,torch_geometric/transform/random_flip.py,RandomFlip,__call__,#RandomFlip#Any#,25

Before Change

    def __call__(self, data):
        if random.random() < self.p:
            return flip(data, self.axis)
        return data

After Change

    def __call__(self, data):
        if random.random() < self.p:
            data.pos[:, self.axis] = -data.pos[:, self.axis]
        return data

    def __repr__(self):

Instances