From 3aa115d867dd2c3cb886cd70bdac8d1d8272426c Mon Sep 17 00:00:00 2001 From: Xinlan Emily Hu Date: Mon, 23 Sep 2024 13:48:52 -0400 Subject: [PATCH] correct issue with updating to the original column --- src/team_comm_tools/feature_builder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/team_comm_tools/feature_builder.py b/src/team_comm_tools/feature_builder.py index 37aa5443..554013ee 100644 --- a/src/team_comm_tools/feature_builder.py +++ b/src/team_comm_tools/feature_builder.py @@ -338,7 +338,7 @@ def __init__( if(not need_sentiment and feature_dict[feature]["bert_sentiment_data"]): need_sentiment = True - check_embeddings(self.chat_data, self.vect_path, self.bert_path, need_sentence, need_sentiment, self.regenerate_vectors, message_col="message_original") + check_embeddings(self.chat_data, self.vect_path, self.bert_path, need_sentence, need_sentiment, self.regenerate_vectors, message_col = self.message_col + "_original") if(need_sentence): self.vect_data = pd.read_csv(self.vect_path, encoding='mac_roman')