diff --git a/model/src/utils/__init__.py b/model/src/utils/__init__.py deleted file mode 100644 index 5f64e61..0000000 --- a/model/src/utils/__init__.py +++ /dev/null @@ -1,3 +0,0 @@ -from __future__ import annotations - -from .check_env import check_env diff --git a/model/src/utils/check_env.py b/model/src/utils/check_env.py deleted file mode 100644 index be2f935..0000000 --- a/model/src/utils/check_env.py +++ /dev/null @@ -1,25 +0,0 @@ -"""Definition of check_env function.""" -from __future__ import annotations - -import os - - -def check_env() -> None: - """Check necessary environment variables are set.""" - necesasary_var_list = { - 'MONGO_HOST', - 'MONGO_DB', - 'MONGO_COLLECTION', - 'MONGO_USER', - 'MONGO_PASSWORD', - 'MINIO_ENDPOINT', - 'MINIO_ACCESS_KEY', - 'MINIO_SECRET_KEY', - } - for env_var in necesasary_var_list: - # ensure env var set - assert ( - env_var in os.environ - ), ( - f"environment variable not set: {env_var}" - ) diff --git a/shared/utils/check_env.py b/shared/utils/check_env.py index c6cf480..99663c3 100644 --- a/shared/utils/check_env.py +++ b/shared/utils/check_env.py @@ -1,4 +1,5 @@ """Definition of check_env function.""" + from __future__ import annotations import os @@ -18,11 +19,8 @@ def check_env() -> None: 'MINIO_ACCESS_KEY', 'MINIO_SECRET_KEY', 'MINIO_BUCKET_NAME', + 'MINIO_BUCKET_NAME_MODELS', } for env_var in necesasary_var_list: # ensure env var set - assert ( - env_var in os.environ - ), ( - f"environment variable not set: {env_var}" - ) + assert env_var in os.environ, f"environment variable not set: {env_var}"