RadiusGraphWithNans Clarification

[vsomnath]

Hi,

Thank you for the great codebase here!

graphqa/src/graphqa/data/graphs.py

Line 90 in 3878efc

na_receiver = torch.any(edge_index[0] == torch.nonzero(nans, as_tuple=False), dim=0)

For my own clarification, shouldn't the line referenced below be:

na_receiver = torch.any(edge_index[1] == torch.nonzero(nans, as_tuple=False), dim=0)

[baldassarreFe]

Hi, that sure looks like it should be edge_index[1]. I don't remember exactly what those two lines do, but it wouldn't make much sense to compare the same thing twice.

Well spotted! Would you be so kind to modify that line, check that it works as intended, and submit a PR?