Skip to content

Commit e00263b

Browse files
committed
Get endpoint_url from s3 env values
1 parent caa3dd7 commit e00263b

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

pipelines/lightweight-components/mobile-price-classifications.ipynb

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,9 +86,13 @@
8686
"# Upload dataset to MinIO\n",
8787
"aws_access_key_id = os.getenv(\"AWS_ACCESS_KEY_ID\", \"\")\n",
8888
"aws_secret_access_key = os.getenv(\"AWS_SECRET_ACCESS_KEY\", \"\")\n",
89+
"s3_endpoint = os.getenv(\"S3_ENDPOINT\", \"minio.minio\")\n",
90+
"s3_use_https = os.getenv(\"S3_USE_HTTPS\", \"0\") == \"1\"\n",
91+
"s3_verify_ssl = os.getenv(\"S3_VERIFY_SSL\", \"0\") == \"1\"\n",
92+
"s3_endpoint = f\"http{'s' if s3_use_https else ''}://{s3_endpoint}\"\n",
8993
"if not aws_access_key_id or not aws_secret_access_key:\n",
9094
" raise ValueError(\"AWS credentials not found in environment variables.\")\n",
91-
"s3 = s3fs.S3FileSystem(endpoint_url=\"http://minio.minio\", key=aws_access_key_id, secret=aws_secret_access_key)\n",
95+
"s3 = s3fs.S3FileSystem(endpoint_url=s3_endpoint, key=aws_access_key_id, secret=aws_secret_access_key, use_ssl=s3_verify_ssl)\n",
9296
"s3_dataset_path = f\"{s3_bucket}/mobile-price-classification\"\n",
9397
"s3.put(f\"{dataset_path}/train.csv\", f\"{s3_dataset_path}/train.csv\")\n",
9498
"s3.put(f\"{dataset_path}/train.csv\", f\"{s3_dataset_path}/test.csv\")\n",

0 commit comments

Comments
 (0)