core/database/__init__.py

@@ -4,3 +4,10 @@ from __future__ import annotations
 from .classes import Dataset
 from .classes import NoDocumentFoundException
 from .classes import VisualCommunication
+from .utils import connect
+from .utils.count_documents import count_documents
+from .utils.get_visual_communication import get_visual_communication
+from .utils.list_names import list_names
+from .utils.upsert_annotation import upsert_annotation
+from .utils.upsert_prediction import upsert_prediction
+from .utils.upsert_visual_communication import upsert_visual_communication

core/database/utils/upsert_annotation.py

@@ -7,7 +7,7 @@ from pymongo.collection import Collection
 from core.dto import ModelData
 
 
-def upsert_annotations(
+def upsert_annotation(
     collection: Collection,
     vis_com_name: str,
     annotations: ModelData,

core/database/utils/upsert_prediction.py

@@ -7,7 +7,7 @@ from pymongo.collection import Collection
 from core.dto import ModelData
 
 
-def upsert_predictions(
+def upsert_prediction(
     collection: Collection,
     vis_com_name: str,
     predictions: ModelData,