correct issue with updating to the original column

Watts-Lab · Sep 23, 2024 · 3aa115d · 3aa115d
1 parent f30bc1d
commit 3aa115d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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')