Create DGLdataset error

[morningphase]

I’m getting this error during runtime:
raise TypeError(default_collate_err_msg_format.format(elem_type))
TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found <class ‘dgl.heterograph.DGLHeteroGraph’>

And the code is like:

class MyDataset(DGLDataset):
    def __init__(self,name=None):
        super(MyDataset, self).__init__(name='dataset')

    def process(self):
        glist, label_dict = load_graphs("preprocessed_graphs/_dgl_graph.bin")
        labels = label_dict['labels'].numpy().tolist()
        self.graphs = glist
        self.label = labels
    def __getitem__(self, idx):
        return self.graphs[idx],th.tensor(self.label[idx])
    def __len__(self):
        return len(self.graphs)

[Rhett-Ying]

please share the whole call stacks. does it happen when creating Dataset or somewhere else using created dataset.

[morningphase]

At that time I was using DataLoader instead of GraphLoader. Now I think there is no problem.