Refactoring ML_Lib

datasets/paired_dataset.py

@@ -6,8 +6,8 @@ import torch
 from torch.utils.data import Dataset, ConcatDataset
 
 from datasets.utils import DatasetMapping
-from lib.preprocessing.generator import Generator
-from lib.objects.map import Map
+from ml_lib.preprocessing.generator import Generator
+from ml_lib.objects.map import Map
 
 
 class TrajPairDataset(Dataset):

datasets/preprocessing/generator.py

@@ -7,7 +7,7 @@ from pathlib import Path
 
 from tqdm import tqdm
 
-from lib.objects.map import Map
+from ml_lib.objects.map import Map
 
 
 class Generator:

datasets/trajectory_dataset.py

@@ -14,11 +14,11 @@ from torch.utils.data import ConcatDataset, Dataset
 import numpy as np
 from tqdm import tqdm
 
-from lib.objects.map import Map
-import lib.variables as V
+from ml_lib.objects.map import Map
+import ml_lib.variables as V
 from PIL import Image
 
-from lib.utils.tools import write_to_shelve
+from ml_lib.utils.tools import write_to_shelve
 
 
 class TrajDataShelve(VisionDataset):