Skip to content

Commit 6f5f982

Browse files
committedMar 17, 2025·
drop pii_analyzed col
1 parent 883298b commit 6f5f982

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed
 

‎utils/export_dataset.py

+9
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,9 @@ def fetch_and_transform_data(table_name, query=None):
8282
if "archived" in df.columns:
8383
df = df.drop(columns=["archived"])
8484

85+
if "pii_analyzed" in df.columns:
86+
df = df.drop(columns=["pii_analyzed"])
87+
8588
# TODO: move 'complete' column from reactions dataset after 'clear_formatting'
8689
# clear_formatting incorrect
8790
# if "complete" in df.columns:
@@ -248,8 +251,14 @@ def main():
248251
columns=["ip_id"]
249252
)
250253

254+
# TODO: refacto in a drop_useless_cols func
255+
251256
if "archived" in df.columns:
252257
df = df.drop(columns=["archived"])
258+
259+
if "pii_analyzed" in df.columns:
260+
df = df.drop(columns=["pii_analyzed"])
261+
253262

254263
export_data(conversations_pii_removed, "conversations")
255264

0 commit comments

Comments
 (0)
Please sign in to comment.