Why is nnz set as edge_index.size(0) in rspmm.cpp?

[cold-blue]

I read the code of rspmm.cpp and find the meaning of nnz should be the whole edge number. So why isn't nnz set as edge_index.size(1)?
Is that an error of code?

I find the last node can't be updated due to this bug.

[migalkin]

Good catch, thanks for noticing - we also stumbled upon the same problem (incorrect node state for the very last node) in another workstream. PR #17 fixes the issue - I ran some tests and now results are correct