diff --git a/src/b2aiprep/demographics.py b/src/b2aiprep/demographics.py index 36345ec..4ab56f6 100644 --- 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 \ No newline at end of file + return output