updated imports

shared/mongodb/__init__.py

@@ -1,9 +1,3 @@
-"""Database module content."""
-
-from __future__ import annotations
-
-from .classes.dataset import Dataset
-from .classes.visual_communication import VisualCommunication
 from .utils.connect_mongodb import connect_mongodb
 from .utils.count_documents import count_documents
 from .utils.get_visual_communication import get_visual_communication

shared/mongodb/classes/visual_communication.py

@@ -13,7 +13,7 @@ from pydantic import BaseModel, ConfigDict
 from pymongo.collection import Collection
 
 from shared.data_store import get, put
-from shared.dto import ModelData
+from shared.mongodb.classes import ModelData
 
 
 class VisualCommunication(BaseModel):

shared/mongodb/utils/get_visual_communication.py

@@ -6,7 +6,7 @@ import logging
 
 from pymongo.collection import Collection
 
-from shared.mongodb import VisualCommunication
+from shared.mongodb.classes import VisualCommunication
 from shared.mongodb.exceptions import NoDocumentFoundException
 
 

shared/mongodb/utils/upsert_annotation.py

@@ -4,7 +4,7 @@ import logging
 
 from pymongo.collection import Collection
 
-from shared.dto import ModelData
+from shared.mongodb.classes import ModelData
 
 
 def upsert_annotation(

shared/mongodb/utils/upsert_prediction.py

@@ -4,7 +4,7 @@ import logging
 
 from pymongo.collection import Collection
 
-from shared.dto import ModelData
+from shared.mongodb.classes import ModelData
 
 
 def upsert_prediction(

shared/mongodb/utils/upsert_visual_communication.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from pymongo.collection import Collection
 
-from shared.mongodb import VisualCommunication
+from shared.mongodb.classes import VisualCommunication
 
 
 def upsert_visual_communication(