mypy fixes

aws-samples · Sep 6, 2024 · cea07f9 · cea07f9
1 parent 162525d
commit cea07f9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/aws_sra_examples/solutions/security_lake/security_lake_org/lambda/src/security_lake.py b/aws_sra_examples/solutions/security_lake/security_lake_org/lambda/src/security_lake.py
@@ -25,6 +25,7 @@
     from mypy_boto3_lakeformation.type_defs import ResourceTypeDef
     from mypy_boto3_organizations import OrganizationsClient
     from mypy_boto3_ram import RAMClient
+    from mypy_boto3_ram.type_defs import ResourceShareInvitationTypeDef
     from mypy_boto3_securitylake import SecurityLakeClient
     from mypy_boto3_securitylake.literals import AwsLogSourceNameType
     from mypy_boto3_securitylake.paginator import ListLogSourcesPaginator
@@ -761,7 +762,7 @@ def configure_resource_share_in_subscriber_acct(ram_client: RAMClient, resource_
         raise ValueError("Error accepting resource share invitation") from None
 
 
-def accept_resource_share_invitation(ram_client: RAMClient, invitation: dict) -> None:
+def accept_resource_share_invitation(ram_client: RAMClient, invitation: ResourceShareInvitationTypeDef) -> None:
     """Accept the resource share invitation.
 
     Args: