brian/visual_critical_discourse_analysis
1
0
Fork 0
Code Issues 11 Pull Requests Actions Projects 1 Releases Activity

updated to fit new workflow

Browse Source
This commit is contained in:
Brian Bjarke Jensen
2024-07-13 23:36:26 +02:00
parent ba1ba027f0
commit a621525d9e
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
shared/data_store/connect.py
+2 -6
View File
@@ -7,7 +7,7 @@ from minio import Minio
def connect( def connect(
bucket: Literal['images', 'models'] = 'images', bucket: Literal['images', 'models'],
) -> Minio: ) -> Minio:
"""Connect to MinIO server.""" """Connect to MinIO server."""
assert isinstance(bucket, str) assert isinstance(bucket, str)
@@ -18,7 +18,6 @@ def connect(
'MINIO_ACCESS_KEY', 'MINIO_ACCESS_KEY',
'MINIO_SECRET_KEY', 'MINIO_SECRET_KEY',
'MINIO_BUCKET_NAME', 'MINIO_BUCKET_NAME',
'MINIO_BUCKET_NAME_MODELS',
] ]
for env_var in necesasary_var_list: for env_var in necesasary_var_list:
# ensure env var set # ensure env var set
@@ -31,10 +30,7 @@ def connect(
minio_endpoint = os.getenv('MINIO_ENDPOINT') minio_endpoint = os.getenv('MINIO_ENDPOINT')
minio_access_key = os.getenv('MINIO_ACCESS_KEY') minio_access_key = os.getenv('MINIO_ACCESS_KEY')
minio_secret_key = os.getenv('MINIO_SECRET_KEY') minio_secret_key = os.getenv('MINIO_SECRET_KEY')
if bucket == 'images': minio_bucket_name = os.getenv('MINIO_BUCKET_NAME')
minio_bucket_name = os.getenv('MINIO_BUCKET_NAME')
else:
minio_bucket_name = os.getenv('MINIO_BUCKET_NAME_MODELS')
# connect client # connect client
client = Minio( client = Minio(
endpoint=minio_endpoint, endpoint=minio_endpoint,