Update src/data/dataset.py

src/data/dataset.py

@@ -314,4 +314,4 @@ class DruggenDataset(InMemoryDataset):
                 if self.pre_transform is not None:
                     data = self.pre_transform(data)
                 data_list.append(data)
-        #torch.save(self.collate(data_list), osp.join(self.processed_dir, self.dataset_file))
+        torch.save(self.collate(data_list), osp.join(self.processed_dir, self.dataset_file))