def clean_cols(df):
    """
    Drop irrelevant columns from the input dataframe
    Args:
        df(pd.DataFrame): Input dataframe

    Returns:
        pd.DataFrame
    """
    snli_df = df.drop(
        [
            "sentence1_binary_parse",
            "sentence2_binary_parse",
            "sentence1_parse",
            "sentence2_parse",
            "captionID",
            "pairID",
            "label1",
            "label2",
            "label3",
            "label4",
            "label5",
        ],
        axis=1,
    )

    snli_df = snli_df.rename(
        columns={"sentence1": S1_COL, "sentence2": S2_COL, "gold_label": LABEL_COL}
    )

    return snli_df

    Drop irrelevant columns from the input dataframe
    Args:
        df(pd.DataFrame): Input dataframe

    Returns:
        pd.DataFrame
    
utils_nlp/dataset/snli.py