Skip to content

Commit

Permalink
Fixed error
Browse files Browse the repository at this point in the history
  • Loading branch information
@akash-vartak
akash-vartak committed Oct 14, 2022
1 parent d72c028 commit 582fc8c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch_geometric/nn/pool/max_pool.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from torch import Tensor
from torch_scatter import scatter

from torch_geometric.data import Batch
from torch_geometric.data import Batch, Data
from torch_geometric.utils import add_self_loops

from .consecutive import consecutive_cluster
Expand Down

0 comments on commit 582fc8c

Please sign in to comment.