fixed mypy types

shared/database/__init__.py

@@ -1,10 +1,11 @@
 """Database module content."""
+
 from __future__ import annotations
 
 from .classes.dataset import Dataset
 from .classes.exceptions import NoDocumentFoundException
 from .classes.visual_communication import VisualCommunication
-from .utils.connect import connect
+from .utils.connect_mongodb import connect_mongodb
 from .utils.count_documents import count_documents
 from .utils.get_visual_communication import get_visual_communication
 from .utils.list_names import list_names

shared/database/utils/save_dataset.py

@@ -20,10 +20,12 @@ def save_dataset(
 
 if __name__ == '__main__':
     from dotenv import load_dotenv
+
     load_dotenv('local.env')
-    from shared.database import list_names, connect
+    from shared.database import connect_mongodb, list_names
+
     # connect to database
-    collection, db, client = connect()
+    collection, db, client = connect_mongodb()
     print(client.server_info())
 
     name_list = list_names(collection=collection, only_with_annotation=True)