def above_the_line(x_array, x1, x2):
    """
    Return states above a specified line defined by (x1, x2).

    We assume that a state has only two coordinates.

    Parameters
    ----------
    x_array: `np.array`
        A 2-d matrix. Usually, an embedding for data points.
    x1: `np.array`
        A list or array of two entries.
    x2: `np.array`
        A list or array of two entries.

    Returns
    -------
    A boolean array.
    """

    return (x_array[:, 1] - x1[1]) > ((x2[1] - x1[1]) / (x2[0] - x1[0])) * (
        x_array[:, 0] - x1[0]
    )