shared/mongodb/__init__.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 from .classes.dataset import Dataset
-from .classes.exceptions import NoDocumentFoundException
 from .classes.visual_communication import VisualCommunication
 from .utils.connect_mongodb import connect_mongodb
 from .utils.count_documents import count_documents

shared/mongodb/classes/__init__.py

@@ -3,5 +3,4 @@
 from __future__ import annotations
 
 from .dataset import Dataset
-from .exceptions import NoDocumentFoundException
 from .visual_communication import VisualCommunication

shared/mongodb/exceptions/__init__.py

@@ -0,0 +1 @@
+from .no_document_found import NoDocumentFoundException

shared/mongodb/exceptions/no_document_found.py

@@ -1,7 +1,5 @@
 """Definition of database exception."""
 
-from __future__ import annotations
-
 
 class NoDocumentFoundException(Exception):
     """Database exception for when no documents are found."""

shared/mongodb/utils/get_visual_communication.py

@@ -6,7 +6,8 @@ import logging
 
 from pymongo.collection import Collection
 
-from shared.mongodb import NoDocumentFoundException, VisualCommunication
+from shared.mongodb import VisualCommunication
+from shared.mongodb.exceptions import NoDocumentFoundException
 
 
 def get_visual_communication(