shared/data_store/connect.py

@@ -7,7 +7,7 @@ from minio import Minio
 
 
 def connect(
-    bucket: Literal['images', 'models'] = 'images',
+    bucket: Literal['images', 'models'],
 ) -> Minio:
     """Connect to MinIO server."""
     assert isinstance(bucket, str)
@@ -18,7 +18,6 @@ def connect(
         'MINIO_ACCESS_KEY',
         'MINIO_SECRET_KEY',
         'MINIO_BUCKET_NAME',
-        'MINIO_BUCKET_NAME_MODELS',
     ]
     for env_var in necesasary_var_list:
         # ensure env var set
@@ -31,10 +30,7 @@ def connect(
     minio_endpoint = os.getenv('MINIO_ENDPOINT')
     minio_access_key = os.getenv('MINIO_ACCESS_KEY')
     minio_secret_key = os.getenv('MINIO_SECRET_KEY')
-    if bucket == 'images':
-        minio_bucket_name = os.getenv('MINIO_BUCKET_NAME')
-    else:
-        minio_bucket_name = os.getenv('MINIO_BUCKET_NAME_MODELS')
+    minio_bucket_name = os.getenv('MINIO_BUCKET_NAME')
     # connect client
     client = Minio(
         endpoint=minio_endpoint,