diff --git a/src/sagemaker/s3.py b/src/sagemaker/s3.py index 57fdef6c06..d81710c412 100644 --- a/src/sagemaker/s3.py +++ b/src/sagemaker/s3.py @@ -110,7 +110,7 @@ def download(s3_uri, local_path, kms_key=None, session=None): sagemaker_session = session or Session() bucket, key_prefix = parse_s3_url(url=s3_uri) if kms_key is not None: - extra_args = {"SSEKMSKeyId": kms_key} + extra_args = {"SSECustomerKey": kms_key} else: extra_args = None diff --git a/tests/unit/test_s3.py b/tests/unit/test_s3.py index ba995ef86f..c073417116 100644 --- a/tests/unit/test_s3.py +++ b/tests/unit/test_s3.py @@ -91,5 +91,5 @@ def test_download_with_kms_key(sagemaker_session): path="/path/for/download/", bucket=BUCKET_NAME, key_prefix=os.path.join(CURRENT_JOB_NAME, SOURCE_NAME), - extra_args={"SSEKMSKeyId": KMS_KEY}, + extra_args={"SSECustomerKey": KMS_KEY}, )