Skip to content

Commit

Permalink
fix docs for lpformer
Browse files Browse the repository at this point in the history
  • Loading branch information
HarryShomer committed Jan 16, 2025
1 parent d992ff2 commit a5e05f2
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions torch_geometric/nn/models/lpformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -369,8 +369,8 @@ def get_ppr_vals(
Args:
batch (Tensor): The batch vector.
Denotes which node pairs to predict.
pair_adj (SparseTensor): Combination of rows in adjacency
for source and target nodes (e.g., X1 + X2)
pair_diff_adj (SparseTensor): Combination of rows in
adjacency for src and tgt nodes (e.g., X1 + X2)
ppr_matrix (Tensor): PPR matrix
"""
# Additional terms for also choosing scores when ppr=0
Expand Down Expand Up @@ -746,9 +746,7 @@ def message(self, x_i: Tensor, x_j: Tensor, ppr_rpes: Tensor,


class MLP(nn.Module):
"""
L Layer MLP.
"""
r"""L Layer MLP."""
def __init__(self, in_channels: int, hid_channels: int, out_channels: int,
num_layers: int = 2, drop: int = 0, norm: str = "layer"):
super().__init__()
Expand Down

0 comments on commit a5e05f2

Please sign in to comment.