Vignesh Ram Somnath

RadiusGraphWithNans Clarification

1

Hi, Thank you for the great codebase here! https://github.com/baldassarreFe/graphqa/blob/3878efc7839bf396a054f4a14e8c031c34f93c40/src/graphqa/data/graphs.py#L90 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)`