def prediction_output_partial(output):
    """
    Function that given the predicted labels, updates the labels to fit intended data output format.
    Specifically, corrects predicted answer spans that start with a 2, to instead start with a 1
    
    Input: array of labels
    Output: array of corrected labels
    """
    corrected_output = []
    prev_label = 0
    for idx, label in enumerate(output):
        if label == 2 and prev_label == 0:
            corrected_output.append(1)
            prev_label = 1
        else:
            corrected_output.append(label)
            prev_label = label
    return corrected_output