diff --git a/shared/data_store/connect.py b/shared/data_store/connect.py index dcc288c..e8aac0f 100644 --- a/shared/data_store/connect.py +++ b/shared/data_store/connect.py @@ -34,14 +34,9 @@ def connect() -> Minio: secret_key=minio_secret_key, secure=False, ) - # ensure buckets exists - bucket_name_list = [ - 'images', - 'models', - ] - 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) + # ensure bucket exists + if not client.bucket_exists(bucket_name=minio_bucket_name): + logging.info('creating bucket: %s', minio_bucket_name) + client.make_bucket(bucket_name=minio_bucket_name) logging.debug('finished') return client