def get_all_logits(
    predictions, features
):
    """After checking assertions against predictions and features length,
    return start & end logtis.

    Args:
        predictions ([Tuple[ndarray, ndarray]]): start & end logit predictions.
        features ([Dataset]): tokenized & splited datasets.

    Returns:
        Tuple([array]): start & end logtis
    """

    assert (
        len(predictions) == 2
    ), "`predictions` should be a tuple with two elements (start_logits, end_logits)."
    all_start_logits, all_end_logits = predictions

    assert len(predictions[0]) == len(
        features
    ), f"Got {len(predictions[0])} predictions and {len(features)} features."

    return all_start_logits, all_end_logits