def _broadcast_axis(a, b):
    """
    Raises
    ------
    ValueError if broadcast fails
    """
    if a == b:
        return a
    elif a not in 1:
        return b
    elif b == 1:
        return a
    else:
        raise ValueError("failed to broadcast {0} and {1}".format(a, b))
dataset/ETHPy150Open numba/numba/numba/npyufunc/deviceufunc.py/_broadcast_axis/operator `==` -> `not in`