diff --git a/misc/generate_random_prediction_test.py b/misc/generate_random_prediction_test.py index 03dda23..edb87f4 100644 --- a/misc/generate_random_prediction_test.py +++ b/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 diff --git a/misc/get_visual_communication_test.py b/misc/get_visual_communication_test.py index 9d84316..725390e 100644 --- a/misc/get_visual_communication_test.py +++ b/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 diff --git a/misc/image_upload_test.py b/misc/image_upload_test.py index 0b979e6..11fa16a 100644 --- a/misc/image_upload_test.py +++ b/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 diff --git a/misc/image_upload_to_server_test.py b/misc/image_upload_to_server_test.py index e3f353c..797e6cb 100644 --- a/misc/image_upload_to_server_test.py +++ b/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 diff --git a/misc/prediction_upload_test.py b/misc/prediction_upload_test.py index 45c9e5a..a6793ae 100644 --- a/misc/prediction_upload_test.py +++ b/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 diff --git a/other/transfer_images_mongo_minio.py b/other/transfer_images_mongo_minio.py index 0b9fcd3..aae76a1 100644 --- a/other/transfer_images_mongo_minio.py +++ b/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