a871536bcbb38b9ca03b0bc777712d8c0a79ad90,matchzoo/data_pack/pack.py,,pack,#Any#,17

Before Change

    // Segment input into 3 dataframes.
    relation = df[col_relation]

    left = df[["id_left", "text_left"]].drop_duplicates(["id_left"])
    left.set_index("id_left", inplace=True)
    // Infer the length of the text left
    // left["length_left"] = left.apply(lambda r: len(r["text_left"]), axis=1)

After Change

    // Build Relation
    relation = pd.DataFrame(data={"id_left": id_left, "id_right": id_right})
    for col in df:
        if col not in ["id_left", "id_right", "text_left", "text_right"]:
            relation[col] = df[col]

    // Build Left and Right
    left = _merge(df, id_left, "text_left", "id_left")
    right = _merge(df, id_right, "text_right", "id_right")
    return matchzoo.DataPack(relation, left, right)

Instances