diff --git a/python/aibrix/aibrix/downloader/s3.py b/python/aibrix/aibrix/downloader/s3.py index c067f904..63cd8025 100644 --- a/python/aibrix/aibrix/downloader/s3.py +++ b/python/aibrix/aibrix/downloader/s3.py @@ -35,9 +35,9 @@ def _parse_bucket_info_from_uri(uri: str) -> Tuple[str, str]: class S3Downloader(BaseDownloader): def __init__(self, model_uri): model_name = envs.DOWNLOADER_MODEL_NAME - ak = envs.DOWNLOADER_AWS_ACCESS_KEY - sk = envs.DOWNLOADER_AWS_SECRET_KEY - endpoint = envs.DOWNLOADER_AWS_ENDPOINT + ak = envs.DOWNLOADER_AWS_ACCESS_KEY_ID + sk = envs.DOWNLOADER_AWS_SECRET_ACCESS_KEY + endpoint = envs.DOWNLOADER_AWS_ENDPOINT_URL region = envs.DOWNLOADER_AWS_REGION bucket_name, bucket_path = _parse_bucket_info_from_uri(model_uri) diff --git a/python/aibrix/aibrix/envs.py b/python/aibrix/aibrix/envs.py index f3742679..8d28889b 100644 --- a/python/aibrix/aibrix/envs.py +++ b/python/aibrix/aibrix/envs.py @@ -70,9 +70,9 @@ def _parse_int_or_none(value: Optional[str]) -> Optional[int]: DOWNLOADER_TOS_ENABLE_CRC = _is_true(os.getenv("TOS_ENABLE_CRC")) # Downloader AWS S3 Envs -DOWNLOADER_AWS_ACCESS_KEY = os.getenv("AWS_ACCESS_KEY_ID") -DOWNLOADER_AWS_SECRET_KEY = os.getenv("AWS_SECRET_ACCESS_KEY") -DOWNLOADER_AWS_ENDPOINT = os.getenv("AWS_ENDPOINT_URL") +DOWNLOADER_AWS_ACCESS_KEY_ID = os.getenv("AWS_ACCESS_KEY_ID") +DOWNLOADER_AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY") +DOWNLOADER_AWS_ENDPOINT_URL = os.getenv("AWS_ENDPOINT_URL") DOWNLOADER_AWS_REGION = os.getenv("AWS_REGION") # Metric Standardizing Related Config