Refactured Settings File

datasets/shapenet.py

@@ -5,16 +5,15 @@ import numpy as np
 from collections import defaultdict
 
 import os
+from torch.utils.data import Dataset
 from tqdm import tqdm
 import glob
 
 import torch
 from torch_geometric.data import InMemoryDataset
 from torch_geometric.data import Data
-from torch.utils.data import Dataset
-import re
 
-from utils.project_config import Classes, DataSplit
+from utils.project_settings import Classes, DataSplit
 
 
 def save_names(name_list, path):
@@ -198,12 +197,13 @@ class ShapeNetPartSegDataset(Dataset):
 
         # y -= 1 if self.num_classes() in y else 0   # Map label from [1, C] to [0, C-1]
 
-        sample = Data(**dict(pos=pos,          # torch.Tensor (n, 3/6)
+        data = Data(**dict(pos=pos,          # torch.Tensor (n, 3/6)
                              y=y,              # torch.Tensor (n,)
                              norm=norm         # torch.Tensor (n, 3/0)
                            )
                       )
-        return sample
+
+        return data
 
     def __len__(self):
         return len(self.dataset)