Skip to content

Commit c36deb5

Browse files
committed
2 parents 8299039 + ece4643 commit c36deb5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

aws_lambda_powertools/utilities/data_masking/providers/aws_encryption_sdk.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
)
1111

1212
from aws_lambda_powertools.utilities.data_masking.provider import Provider
13-
13+
from aws_lambda_powertools.shared.user_agent import register_feature_to_botocore_session
1414

1515
class SingletonMeta(type):
1616
"""Metaclass to cache class instances to optimize encryption"""
@@ -33,7 +33,7 @@ def __call__(cls, *args, **provider_options):
3333
class AwsEncryptionSdkProvider(Provider, metaclass=SingletonMeta):
3434
cache = LocalCryptoMaterialsCache(CACHE_CAPACITY)
3535
session = botocore.session.Session()
36-
36+
register_feature_to_botocore_session(session, "data-masking")
3737
def __init__(self, keys: List[str], client: Optional[EncryptionSDKClient] = None) -> None:
3838
self.client = client or EncryptionSDKClient()
3939
self.keys = keys

0 commit comments

Comments
 (0)