def METHOD_NAME(self, connection: Connection, table_name: str, schema: str = None, **kw) -> List[str]:
    pk = self.get_pk_constraint(connection, table_name, schema)
    return pk.get("constrained_columns")  # type: ignore