'engagement_state' all being 'engaged'

[jkim0731]

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]