fixes

Author: ievgeniia ieromenko

aws_sra_examples/solutions/genai/bedrock_guardrails/lambda/src/sra_config.py

@@ -54,7 +54,7 @@ def get_organization_config_rules(self) -> dict:
         org_id: str = self.ORG_CLIENT.describe_organization()["Organization"]["Id"]
 
         # Get the Organization Config Rules
-        response = self.ORG_CLIENT.describe_organization_config_rules(
+        response = self.ORG_CLIENT.describe_organization_config_rules(  # type: ignore
             OrganizationConfigRuleNames=["sra_config_rule"],
             OrganizationId=org_id,
         )
@@ -75,7 +75,7 @@ def put_organization_config_rule(self) -> dict:
         org_id: str = self.ORG_CLIENT.describe_organization()["Organization"]["Id"]
 
         # Put the Organization Config Rule
-        response = self.ORG_CLIENT.put_organization_config_rule(
+        response = self.ORG_CLIENT.put_organization_config_rule(  # type: ignore
             OrganizationConfigRuleName="sra_config_rule",
             OrganizationId=org_id,
             ConfigRuleName="sra_config_rule",

aws_sra_examples/solutions/genai/bedrock_guardrails/lambda/src/sra_sts.py

@@ -102,13 +102,13 @@ def assume_role(self, account: str, role_name: str, service: str, region_name: s
                 DurationSeconds=900,
             )
             return self.MANAGEMENT_ACCOUNT_SESSION.client(
-                service,
+                service,  # type: ignore
                 region_name=region_name,
                 aws_access_key_id=sts_response["Credentials"]["AccessKeyId"],
                 aws_secret_access_key=sts_response["Credentials"]["SecretAccessKey"],
                 aws_session_token=sts_response["Credentials"]["SessionToken"],
             )
-        return self.MANAGEMENT_ACCOUNT_SESSION.client(service, region_name=region_name, config=self.BOTO3_CONFIG)
+        return self.MANAGEMENT_ACCOUNT_SESSION.client(service, region_name=region_name, config=self.BOTO3_CONFIG)  # type: ignore
 
     def assume_role_resource(self, account: str, role_name: str, service: str, region_name: str) -> Any:
         """Get boto3 resource assumed into an account for a specified service.
@@ -131,7 +131,7 @@ def assume_role_resource(self, account: str, role_name: str, service: str, regio
             DurationSeconds=900,
         )
         return self.MANAGEMENT_ACCOUNT_SESSION.resource(
-            service,
+            service,  # type: ignore
             region_name=region_name,
             aws_access_key_id=sts_response["Credentials"]["AccessKeyId"],
             aws_secret_access_key=sts_response["Credentials"]["SecretAccessKey"],