renamed module

model/src/main.py

@@ -9,7 +9,7 @@ from models import VisualCommunicationModel
 from tqdm import tqdm
 from utils import DEVICE, VCDADataset, load_model
 
-from shared.data_store import connect_minio
+from shared.datastore import connect_minio
 from shared.mongodb.classes import ModelData
 from shared.utils import setup_logging
 

model/src/utils/load_model.py

@@ -7,7 +7,7 @@ import torch
 from minio import Minio
 
 from model.src.models import VisualCommunicationModel
-from shared.data_store import get_model
+from shared.datastore import get_model
 
 from .get_model_name import get_model_name
 

model/src/utils/vcda_dataset.py

@@ -16,7 +16,7 @@ from torchvision.transforms.functional import (
     to_tensor,
 )
 
-from shared.data_store import get_image
+from shared.datastore import get_image
 
 # resnet18 original normalization values
 RESNET_NORMALIZE_MEAN = [0.485, 0.456, 0.406]

model/train.py

@@ -18,7 +18,7 @@ from torch.optim.lr_scheduler import ExponentialLR
 from torch.utils.data import DataLoader
 
 from model.src.utils import VCDADataset, get_class
-from shared.data_store import connect_minio
+from shared.datastore import connect_minio
 
 
 def parse_arguments():