Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Oct 15, 2022
1 parent 3403848 commit 072d8ed
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions torch_geometric/nn/pool/avg_pool.py
Original file line number Diff line number Diff line change
Expand Up @@ -78,8 +78,10 @@ def avg_pool(cluster, data, transform=None):
return data


def avg_pool_neighbor_x(data: Data,
flow: Optional[str] = 'source_to_target', ) -> Data:
def avg_pool_neighbor_x(
data: Data,
flow: Optional[str] = 'source_to_target',
) -> Data:
r"""Average pools neighboring node features, where each feature in
:obj:`data.x` is replaced by the average feature values from the central
node and its neighbors.
Expand Down

0 comments on commit 072d8ed

Please sign in to comment.