skip timestamp col when fill na

Watts-Lab · Feb 28, 2025 · b50ea77 · b50ea77
1 parent 8a352ec
commit b50ea77
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/team_comm_tools/utils/preprocess.py b/src/team_comm_tools/utils/preprocess.py
@@ -71,6 +71,8 @@ def preprocess_conversation_columns(df: pd.DataFrame, column_names: dict, groupi
     df.columns = df.columns.str.replace('[^A-Za-z0-9_]', '', regex=True)
     # fillna
     for role, col in column_names.items():
+        if role == 'timestamp_col':
+            continue
         df[col] = df[col].fillna('')
 
     if not grouping_keys: # case 1: single identifier