model/src/utils/__init__.py

@@ -1,3 +0,0 @@
-from __future__ import annotations
-
-from .check_env import check_env

model/src/utils/check_env.py

@@ -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}"
-        )

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}"