From 312f525a46603f3dd4d6c4715e76eee0858002a7 Mon Sep 17 00:00:00 2001 From: brian Date: Sun, 20 Oct 2024 18:46:32 +0000 Subject: [PATCH] fixed argument for check_env function --- misc/generate_random_prediction_test.py | 3 ++- misc/get_visual_communication_test.py | 3 ++- misc/image_upload_test.py | 3 ++- misc/image_upload_to_server_test.py | 3 ++- misc/prediction_upload_test.py | 3 ++- other/transfer_images_mongo_minio.py | 3 ++- 6 files changed, 12 insertions(+), 6 deletions(-) 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