diff --git a/convokit/model/corpus.py b/convokit/model/corpus.py index 7ec3883e..b24db556 100644 --- a/convokit/model/corpus.py +++ b/convokit/model/corpus.py @@ -1548,9 +1548,10 @@ def from_pandas( df["id"] = df.index # checking if dataframes contain their respective required columns + missing_fields = pd.Index(columns).difference(utterances_df.columns) assert ( - pd.Series(columns).isin(utterances_df.columns).all() - ), "Utterances dataframe must contain all primary data fields" + len(missing_fields) == 0 + ), f"Utterances dataframe is missing the following required fields: {list(missing_fields)}" utterance_meta_cols = extract_meta_from_df(utterances_df)