A small discrepancy between MessagePassing::forward return value and its docstring

[sadra-barikbin]

Hello,
In MessagePassing::forward, it's stated that the shape of return value is [..., n_target_cells, out_channels] but its return value is actually of shape [..., n_target_cells_with_neighbors, out_channels] which its last but one dimension size is the number of target cells that have any neighbor in source cells. Which one is supposed to be correct? The implementation or the docstring?

To reproduce:

from topomodelx.base.message_passing import MessagePassing
m = MessagePassing()
x_source = torch.rand((3,10))
neighborhood = torch.tensor([[1, 0, 1], [0, 0, 0]]).to_sparse_coo()
m.forward(x_source, neighborhood).shape # result: torch.Size([1, 10])