diff --git a/torch_geometric/nn/models/lpformer.py b/torch_geometric/nn/models/lpformer.py
index 711a9b1e3a51..b44223e99b79 100644
--- a/torch_geometric/nn/models/lpformer.py
+++ b/torch_geometric/nn/models/lpformer.py
@@ -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
@@ -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__()