Update demographics.py

sensein · Apr 19, 2024 · ec1301a · ec1301a
1 parent 8846920
commit ec1301a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/b2aiprep/demographics.py b/src/b2aiprep/demographics.py
@@ -103,7 +103,7 @@ def get_columns_of_repeat_instrument(repeat_instrument: RepeatInstrument) -> Lis
 
 def get_df_of_repeat_instrument(df: DataFrame, repeat_instrument: RepeatInstrument) -> pd.DataFrame:
     columns = get_columns_of_repeat_instrument(repeat_instrument)
-    if repeat_instrument in (RepeatInstrument.GENERIC_DEMOGRAPHICS, RepeatInstrument.PARTICIPANT):
+    if repeat_instrument in (RepeatInstrument.PARTICIPANT,):
         idx = df['redcap_repeat_instrument'].isnull()
     else:
         idx = df['redcap_repeat_instrument'] == repeat_instrument.value
@@ -143,4 +143,4 @@ def load_features_for_recordings(df: pd.DataFrame, data_path: Path, feature: Opt
         if feature is not None:
             output[recording_id] = output[recording_id][feature]
 
-    return output
+    return output