Degree-Quant dataset transform bug

[chaitjo]

Thank you for releasing the code! Found a bug in how transforms are composed for Degree-Quant when I customized the REDDIT example code to my own dataset:

degree-quant/reddit_binary/dataset.py

Line 71 in 5f05601

dataset.transform = T.Compose([dataset.transform, dq_transform])

Had to change to:

if dataset.transform is None:
    dataset.transform = dq_transform
else:
    dataset.transform = T.Compose([dataset.transform, dq_transform])

Otherwise, the transform becomes a list [None, ProbabilisticHighDegreeMask].

[chaitjo]

P.S. This doesn't apply for REDDIT_BINARY. This applies to datasets with rich node attributes.

[shyam196]

I added a comment for others, thanks for pointing out :-)