Update typo in doc-string (#9057)

Closes #9056
pyg-team · Mar 18, 2024 · 18d38c9 · 18d38c9
1 parent 12f6dcb
commit 18d38c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric/nn/aggr/base.py b/torch_geometric/nn/aggr/base.py
@@ -47,7 +47,7 @@ class Aggregation(torch.nn.Module):
        # Define the boundary indices for three sets:
        ptr = torch.tensor([0, 4, 7, 10])
 
-       output = aggr(x, ptr=ptr)  #  Output shape: [4, 64]
+       output = aggr(x, ptr=ptr)  #  Output shape: [3, 64]
 
     Note that at least one of :obj:`index` or :obj:`ptr` must be defined.