Creating an InMemoryDataset with HeteroData

[mdanb]

I have HeteroData which I'm using in an InMemoryDataset. From the docs, I need to save the data in process as:

data, slices = self.collate([data])
torch.save((data, slices), self.processed_paths[0])

However, it seems like collate doesn't work with HeteroData, only Data. So what's the best solution here?

[mdanb]

Actually, I was able to collate even though it expects Data

[rusty1s]

Yes, it should support both Data and HeteroData. Let me update the type hints.