diff --git a/dataset/shapenet.py b/dataset/shapenet.py
index c8a694e..33f4dbf 100644
--- a/dataset/shapenet.py
+++ b/dataset/shapenet.py
@@ -1,25 +1,15 @@
-import os
 import numpy as np
 
-from torch.utils.data import Dataset
-from torch_geometric.datasets import ShapeNet
-
-from itertools import repeat, product
 from collections import defaultdict
 
 import os
 from tqdm import tqdm
-import os.path as osp
 import glob
 
 import torch
-from torch_geometric.data import (Data, InMemoryDataset, download_url,
-                                  extract_zip)
+from torch_geometric.data import InMemoryDataset
+from torch_geometric.data import Data
 from torch.utils.data import Dataset
-from torch_geometric.read import read_txt_array
-from torch_geometric.datasets import ShapeNet
-
-from torch_geometric.read import parse_txt_array
 
 
 class CustomShapeNet(InMemoryDataset):