Skip to content

Commit

Permalink
Change torch_scatter.scatter to utils.scatter
Browse files Browse the repository at this point in the history
  • Loading branch information
HarryShomer committed Jan 16, 2025
1 parent c5005f1 commit c2bc98f
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions torch_geometric/nn/models/lpformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,13 +3,12 @@

import torch
import torch.nn as nn
from torch_scatter import scatter
import torch.nn.functional as F
from torch import Tensor
from torch.nn import Parameter
from torch_geometric.typing import SparseTensor

from ...utils import softmax, get_ppr
from ...utils import softmax, get_ppr, scatter
from ...typing import OptTensor, Tuple, Adj
from ...nn.conv import MessagePassing
from ...nn.dense.linear import Linear
Expand Down

0 comments on commit c2bc98f

Please sign in to comment.