def create_adv_inputs_single(states):
    """
    Create the input for the ADV Net: cab-positions + psng-positions
    """
    # For testing this only works for 2 agents and 2 passengers
    assert len(states) == 2
    assert len(states[0]) == 9

    # passenger is the same for all cabs
    pass1_x, pass1_y = states[0][7], states[0][8]

    cab1_pos_x, cab1_pos_y = states[0][5], states[0][6]
    cab2_pos_x, cab2_pos_y = states[1][5], states[1][6]

    return [cab1_pos_x, cab1_pos_y, cab2_pos_x, cab2_pos_y, pass1_x, pass1_y]