Fix mypy errors

seshubaws · Nov 14, 2023 · 65a9e9f · 65a9e9f
1 parent 1609105
commit 65a9e9f
Show file tree

Hide file tree

Showing 6 changed files with 10 additions and 10 deletions.
diff --git a/examples/data_masking/src/custom_provider.py b/examples/data_masking/src/custom_provider.py
@@ -1,6 +1,6 @@
 from itsdangerous.url_safe import URLSafeSerializer
 
-from aws_lambda_powertools.utilities.data_masking.provider import BaseProvider
+from aws_lambda_powertools.utilities._data_masking.provider import BaseProvider
 
 
 class MyCustomEncryption(BaseProvider):

diff --git a/examples/data_masking/src/data_masking_function_example.py b/examples/data_masking/src/data_masking_function_example.py
@@ -1,8 +1,8 @@
 import os
 
 from aws_lambda_powertools import Logger, Tracer
-from aws_lambda_powertools.utilities.data_masking import DataMasking
-from aws_lambda_powertools.utilities.data_masking.provider.kms.aws_encryption_sdk import AwsEncryptionSdkProvider
+from aws_lambda_powertools.utilities._data_masking import DataMasking
+from aws_lambda_powertools.utilities._data_masking.provider.kms.aws_encryption_sdk import AwsEncryptionSdkProvider
 from aws_lambda_powertools.utilities.typing import LambdaContext
 
 KMS_KEY_ARN = os.environ["KMS_KEY_ARN"]

diff --git a/examples/data_masking/src/getting_started_encrypt_data.py b/examples/data_masking/src/getting_started_encrypt_data.py
@@ -1,7 +1,7 @@
 import os
 
-from aws_lambda_powertools.utilities.data_masking import DataMasking
-from aws_lambda_powertools.utilities.data_masking.provider.kms.aws_encryption_sdk import AwsEncryptionSdkProvider
+from aws_lambda_powertools.utilities._data_masking import DataMasking
+from aws_lambda_powertools.utilities._data_masking.provider.kms.aws_encryption_sdk import AwsEncryptionSdkProvider
 
 KMS_KEY_ARN = os.environ["KMS_KEY_ARN"]
 

diff --git a/examples/data_masking/src/getting_started_mask_data.py b/examples/data_masking/src/getting_started_mask_data.py
@@ -1,4 +1,4 @@
-from aws_lambda_powertools.utilities.data_masking import DataMasking
+from aws_lambda_powertools.utilities._data_masking import DataMasking
 
 
 def lambda_handler(event, context):

diff --git a/examples/data_masking/src/working_with_own_provider.py b/examples/data_masking/src/working_with_own_provider.py
@@ -1,4 +1,4 @@
-from aws_lambda_powertools.utilities.data_masking.base import DataMasking
+from aws_lambda_powertools.utilities._data_masking.base import DataMasking
 from examples.data_masking.src.custom_provider import MyCustomEncryption
 
 

diff --git a/examples/data_masking/tests/src/single_mock.py b/examples/data_masking/tests/src/single_mock.py
@@ -3,9 +3,9 @@
 import base64
 from typing import Any, Callable, Dict, Union
 
-from aws_lambda_powertools.utilities.data_masking.base import DataMasking
-from aws_lambda_powertools.utilities.data_masking.provider import BaseProvider
-from aws_lambda_powertools.utilities.data_masking.provider.kms.aws_encryption_sdk import AwsEncryptionSdkProvider
+from aws_lambda_powertools.utilities._data_masking.base import DataMasking
+from aws_lambda_powertools.utilities._data_masking.provider import BaseProvider
+from aws_lambda_powertools.utilities._data_masking.provider.kms.aws_encryption_sdk import AwsEncryptionSdkProvider
 
 
 class FakeEncryptionKeyProvider(BaseProvider):