If I am correct, it may be worth adding a comment on the need to remove self-loops in careful pre-processing instead of using the default functionality in the HeteroSAGEConv layer? Another option could be to add a check if the source and destination node type are the same in the Hetero...