-
Notifications
You must be signed in to change notification settings - Fork 7
Description
All engagement_state values are 'engaged' regardless of values on engaged column in extended_stimulus_presentations and extended_trials tables.
Checked by
dataset = cache.get_behavior_ophys_experiment(ophys_experiment_id)
extended_stimulus = loading.get_extended_stimulus_presentations_table(dataset.stimulus_presentations, dataset.licks, dataset.rewards, dataset.running_speed)
extended_stimulus['reward_rate','engaged','engagement_state'].head(50)
A simple fix might work from
def get_extended_stimulus_presentations_table in loading.py
where
stimulus_presentations['engagement_state'] = ['engaged' if True else 'disengaged' for engaged in stimulus_presentations['engaged'].values]
should have been
stimulus_presentations['engagement_state'] = ['engaged' if engaged==True else 'disengaged' for engaged in stimulus_presentations['engaged'].values]