def if_cond(switch, size, num):
    if (switch > 0.5):
        return [size, num, num]
    return [size, num]
def if_cond(switch, size, num):
    if (switch > 0.5):
        return [size, num, num]
    return [size, num]