Bugs on TUDatasets with no features

[zekarias-tilahun]

Hi,

Thank you very much for an excellent work. I have been trying your work on TUDatasets and it works well with the chemical datasets, however, for those dataset with no features (e.g., IMDB-BINARY, IMDB-MULTI, REDDIT-BINARY, COLLAB) I encountered a bug.
I've tried to resolve it for some time now and failed to do so. I very much appreciate it if you could let me know
(1) if you have tried your code on the aforementioned datasets and worked for you
(2) if not, would you mind looking at the trace and give me your feedback.

Traceback (most recent call last):
  File "~/GraphCL_Automated/unsupervised_TU/joaov2.py", line 158, in <module>
    print(dataset.get_num_feature())
  File "~/GraphCL_Automated/unsupervised_TU/aug.py", line 189, in get_num_feature
    item, slices = self.data[key], self.slices[key]
KeyError: 'num_nodes'

Thank you in advance for your help!

[yyou1996]

Hi @zekarias-tilahun,

Please refer to Shen-Lab/GraphCL#36 (comment).

[zekarias-tilahun]

That you for the quick reply @yyou1996, that has fixed the problem.