5cd8f0e4aa0a74122eed266dbf1f657dc833366b,torch_geometric/datasets/dataset.py,Dataset,_process,#Dataset#,82

Before Change

    def _process(self):
        filename = self._processed_file
        if exists(filename):
            self.datas = torch.load(filename)
            return

        self.datas = self.process()
        torch.save(to_list(self.datas), filename)

After Change

        sets = sets if isinstance(sets, tuple) else (sets, )

        // Save training and test set separately.
        for i in range(len(self._processed_files)):
            torch.save(to_list(sets[i]), self._processed_files[i])

Instances