def get_target(model, observation):
    """Get target from observation."""
    if model.model_kind == "dynamics":
        target = observation.next_state
    elif model.model_kind == "rewards":
        target = observation.reward
    elif model.model_kind == "termination":
        target = observation.done
    else:
        raise NotImplementedError
    return target