From 4740e94d37035f4b7cab9ca684d252a887b993c8 Mon Sep 17 00:00:00 2001 From: junos Date: Wed, 18 Aug 2021 10:48:34 +0200 Subject: [PATCH] Fix a bug introduced in e7fe4e8398c3440aa940ba93fbddd797ce473797 . --- features/communication.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/features/communication.py b/features/communication.py index 87594fb..2c8b42b 100644 --- a/features/communication.py +++ b/features/communication.py @@ -167,8 +167,8 @@ def count_comms(comm_df: pd.DataFrame) -> pd.DataFrame: comm_contacts_counts = ( enumerate_contacts(comm_df) .groupby(["participant_id"]) - .nunique() - .rename(columns={"contact_id": "no_contacts"}) + .nunique()["contact_id"] + .rename("no_contacts") ) # Number of communication contacts comm_features = comm_counts.join(comm_duration_total) @@ -199,8 +199,8 @@ def count_comms(comm_df: pd.DataFrame) -> pd.DataFrame: comm_contacts_counts = ( enumerate_contacts(comm_df) .groupby(["participant_id"]) - .nunique() - .rename(columns={"contact_id": "no_contacts"}) + .nunique()["contact_id"] + .rename("no_contacts") ) # Number of communication contacts comm_features = comm_features.join(comm_contacts_counts)