updated imports

tests/prediction_upload_test.py

@@ -4,11 +4,12 @@ import logging
 import os
 from pathlib import Path
 
-from database import connect
-from database import upsert_predictions
-from database import VisualCommunication
 from dotenv import load_dotenv
 
+from core.database import connect
+from core.database import upsert_predictions
+from core.database.classes import VisualCommunication
+
 if __name__ == '__main__':
     # setup logging
     fmt = (