diff --git a/shared/data_store/connect.py b/shared/data_store/connect.py index 14c61ae..dcc288c 100644 --- a/shared/data_store/connect.py +++ b/shared/data_store/connect.py @@ -2,25 +2,20 @@ import logging import os -from typing import Literal from minio import Minio -def connect( - bucket: Literal['images', 'models'], -) -> Minio: +def connect() -> Minio: """Connect to MinIO server.""" - assert isinstance(bucket, str) - assert bucket in ['images', 'models'] # ensure necessary env vars available - necesasary_var_list = [ + env_var_list = [ 'MINIO_ENDPOINT', 'MINIO_ACCESS_KEY', 'MINIO_SECRET_KEY', 'MINIO_BUCKET_NAME', ] - for env_var in necesasary_var_list: + for env_var in env_var_list: # ensure env var set assert ( env_var in os.environ @@ -40,11 +35,11 @@ def connect( secure=False, ) # ensure buckets exists - bucket_list = [ + bucket_name_list = [ 'images', 'models', ] - for bucket_name in bucket_list: + for bucket_name in bucket_name_list: if not client.bucket_exists(bucket_name=bucket_name): logging.info('creating bucket: %s', bucket_name) client.make_bucket(bucket_name=minio_bucket_name)