fix: mypy

aws · Nov 4, 2024 · 311afc8 · 311afc8
1 parent c8424cf
commit 311afc8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/awswrangler/data_quality/_create.py b/awswrangler/data_quality/_create.py
@@ -170,7 +170,7 @@ def update_ruleset(
         df_existing = get_ruleset(name=name, boto3_session=boto3_session)
         df_existing = df_existing.set_index(keys=["rule_type", "parameter"], drop=False, verify_integrity=True)
         df_updated = _rules_to_df(dqdl_rules) if dqdl_rules is not None else df_rules
-        df_updated = df_updated.set_index(keys=["rule_type", "parameter"], drop=False, verify_integrity=True)
+        df_updated = df_updated.set_index(keys=["rule_type", "parameter"], drop=False, verify_integrity=True)  # type: ignore[union-attr]
         merged_df = pd.concat([df_existing[~df_existing.index.isin(df_updated.index)], df_updated])
         dqdl_rules = _create_dqdl(merged_df.reset_index(drop=True))
     else:

diff --git a/test_infra/stacks/base_stack.py b/test_infra/stacks/base_stack.py
@@ -30,7 +30,7 @@ def __init__(self, scope: Construct, construct_id: str, **kwargs: str) -> None:
 
         # Add outputs for the EIPs generated by the VPC construct
         for public_subnet in self.vpc.public_subnets:
-            elastic_ip: ec2.CfnEIP = public_subnet.node.find_child("EIP")  # type:ignore
+            elastic_ip: ec2.CfnEIP = public_subnet.node.find_child("EIP")  # type: ignore
             ssm.StringParameter(
                 self,
                 f"{public_subnet.node.id} EIP SSM",