core/database/__init__.py

@@ -1,11 +1,6 @@
+"""Database module content."""
 from __future__ import annotations
 
-from .base import connect
+from .classes import Dataset
 from .classes import NoDocumentFoundException
 from .classes import VisualCommunication
-from .utils import count_documents
-from .utils import get_visual_communication
-from .utils import list_names
-from .utils import upsert_annotations
-from .utils import upsert_predictions
-from .utils import upsert_visual_communication

core/database/utils/__init__.py

@@ -1,8 +1,2 @@
+"""Database utils module content."""
 from __future__ import annotations
-
-from .count_documents import count_documents
-from .get_visual_communication import get_visual_communication
-from .list_names import list_names
-from .upsert_annotation import upsert_annotations
-from .upsert_prediction import upsert_predictions
-from .upsert_visual_communication import upsert_visual_communication