From a621525d9e06fbf13be8d7f87b724ba823a68893 Mon Sep 17 00:00:00 2001 From: Brian Bjarke Jensen Date: Sat, 13 Jul 2024 23:36:26 +0200 Subject: [PATCH] updated to fit new workflow --- shared/data_store/connect.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/shared/data_store/connect.py b/shared/data_store/connect.py index 7e102e8..ddf4c41 100644 --- a/shared/data_store/connect.py +++ b/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,