Added who_participates column

LAAC-LSCP · Mar 14, 2024 · f47e5aa · f47e5aa
1 parent d3c06cd
commit f47e5aa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/ChildProject/pipelines/conversationFunctions.py b/ChildProject/pipelines/conversationFunctions.py
@@ -92,6 +92,9 @@ def who_initiated(annotations: pd.DataFrame):
 def who_finished(annotations: pd.DataFrame):
     return annotations.reset_index().iloc[-1]['speaker_type']
 
+@conversationFunction()
+def who_participates(annotations: pd.DataFrame):
+    return annotations.reset_index()['speaker_type'].unique()
 
 @conversationFunction()
 def total_duration_of_vocalisations(annotations: pd.DataFrame):

diff --git a/ChildProject/pipelines/conversations.py b/ChildProject/pipelines/conversations.py
@@ -401,6 +401,7 @@ def __init__(
              ["vocalisations_count", "vocalisations_count", pd.NA],
              ["who_initiated", "initiator", pd.NA],
              ["who_finished", "finisher", pd.NA],
+             ["who_participates", "participators", pd.NA],
              ["total_duration_of_vocalisations", "total_duration_of_vocalisations", pd.NA],
              ["conversation_duration", "conversation_duration", pd.NA],
              ["is_speaker", "CHI_present", 'CHI'],