def get_aws_partition(context):
    """
    Get aws partition

    @param context: the context of the event
    @return: current account partition
    """
    return context.invoked_function_arn.split(":")[1]