def unique_values_from_query(query_result):
    """Simplify Athena query results into a set of values.

    Useful for listing tables, partitions, databases, enable_metrics

    Args:
        query_result (dict): The result of run_athena_query

    Returns:
        set: Unique values from the query result
    """
    return {
        value
        for row in query_result['ResultSet']['Rows'] for result in row['Data']
        for value in list(result.values())
    }