ogbg-molchembl dataset fails due to default pickle_protocol about ogb HOT 4 CLOSED

This does not occur with the dgl variant.

from ogb.

Interesting. I tested the following locally, and it worked fine. Could you clarify what you mean by "This does not occur with the dgl variant."?

from ogb.graphproppred import DglGraphPropPredDataset
    
d_name = 'ogbg-molchembl'
dataset = DglGraphPropPredDataset(name = d_name) 

from ogb.

Ah I mean the Dgl variant works, and it's the pure python dataset which fails to save the pickle file. I think the further processing to Dgl datastructure reduces size of the pickle enough.

Edited root comment to reflect it's GraphPropPredDataset that fails.

from ogb.

You are right, thanks for noticing this. I have resolved the issue in the master branch by using protocol = 4 in torch.save().

from ogb.