Batching a nested list of graphs in torch_geometric

I am currently training a model which is a mix of graph neural networks and LSTM. However that means for each of my training sample, I need to pass in a list of graphs. The current batch class in torch_geometric supports batching with but this only allows one graph for each data point. How else can I go about batching the graphs when each data consist of multiple graphs?

Do you want to post the question/request to torch_geometric repo? We are not familiar with that.

ah I didn’t think of that, thanks for the suggestion!