updates to checking columns are in data

amytangzheng · amytangzheng · commit 3f31f07ee164 · 2024-11-08T14:14:10.000-05:00
diff --git a/src/team_comm_tools/utils/calculate_conversation_level_features.py b/src/team_comm_tools/utils/calculate_conversation_level_features.py
@@ -98,7 +98,7 @@ def __init__(self, chat_data: pd.DataFrame,
                 convo_columns_in_data = list(set(convo_columns).intersection(set(self.chat_data.columns)))
                 if(len(convo_columns_in_data) != len(convo_columns)):
                     print(
-                        "Warning: One or more requested user columns are not present in the data. Ignoring them."
+                        "Warning: One or more requested user columns are not present in the DATA. Ignoring them."
                     )
                     
                     for i in convo_columns:
@@ -116,7 +116,7 @@ def __init__(self, chat_data: pd.DataFrame,
                 convo_columns_in_features = list(set(convo_columns).intersection(set(self.chat_features)))
                 if(len(convo_columns_in_features) != len(convo_columns)):
                     print(
-                        "Warning: One or more requested user columns are not present in the feature list. Ignoring them."
+                        "Warning: One or more requested user columns are not present in the FEATURE LIST. Ignoring them."
                     )
                     
                     for i in convo_columns:
diff --git a/src/team_comm_tools/utils/calculate_user_level_features.py b/src/team_comm_tools/utils/calculate_user_level_features.py
@@ -69,7 +69,7 @@ def __init__(self, chat_data: pd.DataFrame,
                 
                 if(len(user_columns_in_data) != len(user_columns)):
                     print(
-                        "Warning: One or more requested user columns are not present in the data. Ignoring them."
+                        "Warning: One or more requested user columns are not present in the DATA. Ignoring them."
                     )
                     
                     for i in user_columns:
@@ -88,7 +88,7 @@ def __init__(self, chat_data: pd.DataFrame,
                 
                 if(len(user_columns_in_features) != len(user_columns)):
                     print(
-                        "Warning: One or more requested user columns are not present in the feature list. Ignoring them."
+                        "Warning: One or more requested user columns are not present in the FEATURE LIST. Ignoring them."
                     )
                     
                     for i in user_columns:

Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,7 @@ def __init__(self, chat_data: pd.DataFrame,`
`69`	`69`
`70`	`70`	`if(len(user_columns_in_data) != len(user_columns)):`
`71`	`71`	`print(`
`72`		`- "Warning: One or more requested user columns are not present in the data. Ignoring them."`
	`72`	`+ "Warning: One or more requested user columns are not present in the DATA. Ignoring them."`
`73`	`73`	`)`
`74`	`74`
`75`	`75`	`for i in user_columns:`
`@@ -88,7 +88,7 @@ def __init__(self, chat_data: pd.DataFrame,`
`88`	`88`
`89`	`89`	`if(len(user_columns_in_features) != len(user_columns)):`
`90`	`90`	`print(`
`91`		`- "Warning: One or more requested user columns are not present in the feature list. Ignoring them."`
	`91`	`+ "Warning: One or more requested user columns are not present in the FEATURE LIST. Ignoring them."`
`92`	`92`	`)`
`93`	`93`
`94`	`94`	`for i in user_columns:`