cr fixes

Author: Ran Isenberg

aws_lambda_powertools/utilities/parameters/appconfig.py

@@ -4,12 +4,14 @@
 
 
 import os
-from typing import Any, Dict, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, Optional, Union
 from uuid import uuid4
 
 import boto3
 from botocore.config import Config
-from mypy_boto3_appconfig import AppConfigClient
+
+if TYPE_CHECKING:
+    from mypy_boto3_appconfig import AppConfigClient
 
 from ...shared import constants
 from ...shared.functions import resolve_env_var_choice
@@ -71,7 +73,7 @@ def __init__(
         application: Optional[str] = None,
         config: Optional[Config] = None,
         boto3_session: Optional[boto3.session.Session] = None,
-        boto3_client: Optional[AppConfigClient] = None,
+        boto3_client: Optional["AppConfigClient"] = None,
     ):
         """
         Initialize the App Config client

aws_lambda_powertools/utilities/parameters/secrets.py

@@ -3,11 +3,13 @@
 """
 
 
-from typing import Any, Dict, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, Optional, Union
 
 import boto3
 from botocore.config import Config
-from mypy_boto3_secretsmanager import SecretsManagerClient
+
+if TYPE_CHECKING:
+    from mypy_boto3_secretsmanager import SecretsManagerClient
 
 from .base import DEFAULT_MAX_AGE_SECS, DEFAULT_PROVIDERS, BaseProvider
 
@@ -67,7 +69,7 @@ def __init__(
         self,
         config: Optional[Config] = None,
         boto3_session: Optional[boto3.session.Session] = None,
-        boto3_client: Optional[SecretsManagerClient] = None,
+        boto3_client: Optional["SecretsManagerClient"] = None,
     ):
         """
         Initialize the Secrets Manager client

aws_lambda_powertools/utilities/parameters/ssm.py

@@ -3,14 +3,16 @@
 """
 
 
-from typing import Any, Dict, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, Optional, Union
 
 import boto3
 from botocore.config import Config
-from mypy_boto3_ssm import SSMClient
 
 from .base import DEFAULT_MAX_AGE_SECS, DEFAULT_PROVIDERS, BaseProvider
 
+if TYPE_CHECKING:
+    from mypy_boto3_ssm import SSMClient
+
 
 class SSMProvider(BaseProvider):
     """
@@ -83,7 +85,7 @@ def __init__(
         self,
         config: Optional[Config] = None,
         boto3_session: Optional[boto3.session.Session] = None,
-        boto3_client: Optional[SSMClient] = None,
+        boto3_client: Optional["SSMClient"] = None,
     ):
         """
         Initialize the SSM Parameter Store client

poetry.lock

@@ -27,7 +27,6 @@ boto3 = "^1.18"
 jmespath = "^0.10.0"
 pydantic = {version = "^1.8.2", optional = true }
 email-validator = {version = "*", optional = true }
-boto3-stubs = {extras = ["appconfig", "ssm", "secretsmanager"], version = "^1.21.37"}
 
 [tool.poetry.dev-dependencies]
 coverage = {extras = ["toml"], version = "^6.2"}
@@ -55,6 +54,9 @@ mkdocs-git-revision-date-plugin = "^0.3.1"
 mike = "^0.6.0"
 mypy = "^0.931"
 mkdocs-material = "^8.2.4"
+mypy-boto3-appconfig = "^1.21.34"
+mypy-boto3-ssm = "^1.21.34"
+mypy-boto3-secretsmanager = "^1.21.34"
 
 
 [tool.poetry.extras]