51b53dcbab8ec7ab0b6e8a64284a919db2d2254a,torch_geometric/transform/local_cartesian.py,LocalCartesian,__call__,#LocalCartesian#Any#,8

Before Change

        if data.weight is None:
            data.weight = weight
        else:
            data.weight = torch.cat([weight, data.weight.unsqueeze(1)], dim=1)

        return data

After Change

        cartesian = pos[col] - pos[row]
        tmp, _ = scatter_max(cartesian.abs(), row, 0, dim_size=pos.size(0))
        cartesian /= 2 * tmp.max(dim=1, keepdim=True)[0][row]
        cartesian += 0.5

        if pseudo is not None and self.cat:
            pseudo = pseudo.view(-1, 1) if pseudo.dim() == 1 else pseudo
            cartesian = cartesian.type_as(pseudo)
            data.weight = torch.cat([pseudo, cartesian], dim=-1)
        else:
            data.weight = cartesian

Instances