misc/generate_random_prediction_test.py

@@ -8,6 +8,7 @@ from shared.data_store import connect_minio
 from shared.database import connect_mongodb
 from shared.database.classes import VisualCommunication
 from shared.utils import check_env, setup_logging
+from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 
 if __name__ == '__main__':
     # load in env file
@@ -15,7 +16,7 @@ if __name__ == '__main__':
     assert env_path.exists()
     load_dotenv(env_path)
     # ensure env vars set
-    check_env()
+    check_env(NECESSARY_ENV_VAR_LIST)
     # setup logging
     setup_logging()
     # connect to minIO

misc/get_visual_communication_test.py

@@ -7,6 +7,7 @@ from dotenv import load_dotenv
 from shared.data_store import connect_minio
 from shared.database import connect_mongodb, get_visual_communication
 from shared.utils import check_env, setup_logging
+from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 
 if __name__ == '__main__':
     # load in env file
@@ -14,7 +15,7 @@ if __name__ == '__main__':
     assert env_path.exists()
     load_dotenv(env_path)
     # ensure env vars set
-    check_env()
+    check_env(NECESSARY_ENV_VAR_LIST)
     # setup logging
     setup_logging()
     # connect to minIO

misc/image_upload_test.py

@@ -9,6 +9,7 @@ from shared.data_store import connect_minio
 from shared.database import connect_mongodb
 from shared.database.classes import VisualCommunication
 from shared.utils import check_env, setup_logging
+from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 
 if __name__ == '__main__':
     # load in env file
@@ -16,7 +17,7 @@ if __name__ == '__main__':
     assert env_path.exists()
     load_dotenv(env_path)
     # ensure env vars set
-    check_env()
+    check_env(NECESSARY_ENV_VAR_LIST)
     # setup logging
     setup_logging()
     # connect to minIO

misc/image_upload_to_server_test.py

@@ -8,6 +8,7 @@ from pymongo.errors import DuplicateKeyError
 from shared.data_store import connect_minio
 from shared.database import VisualCommunication, connect_mongodb
 from shared.utils import check_env, setup_logging
+from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 
 if __name__ == '__main__':
     # load in env file
@@ -15,7 +16,7 @@ if __name__ == '__main__':
     assert env_path.exists()
     load_dotenv(env_path)
     # ensure env vars set
-    check_env()
+    check_env(NECESSARY_ENV_VAR_LIST)
     # setup logging
     setup_logging()
     # connect to minIO

misc/prediction_upload_test.py

@@ -8,6 +8,7 @@ from shared.data_store import connect_minio
 from shared.database import connect_mongodb, upsert_prediction
 from shared.database.classes import VisualCommunication
 from shared.utils import check_env, setup_logging
+from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 
 if __name__ == '__main__':
     # load in env file
@@ -15,7 +16,7 @@ if __name__ == '__main__':
     assert env_path.exists()
     load_dotenv(env_path)
     # ensure env vars set
-    check_env()
+    check_env(NECESSARY_ENV_VAR_LIST)
     # setup logging
     setup_logging()
     # connect to minIO

other/transfer_images_mongo_minio.py

@@ -13,6 +13,7 @@ from pymongo.collection import Collection
 from shared.data_store import connect_minio, put
 from shared.database import VisualCommunication, connect_mongodb
 from shared.utils import check_env, setup_logging
+from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 
 
 def list_mongo_document_ids(
@@ -76,7 +77,7 @@ if __name__ == '__main__':
     assert env_path.exists()
     load_dotenv(env_path)
     # ensure env vars set
-    check_env()
+    check_env(NECESSARY_ENV_VAR_LIST)
     # setup logging
     setup_logging()
     # connect to minIO