def get_trials(self):
licks = self.get_licks()
behavior_stimulus_file = self.get_behavior_stimulus_file()
data = pd.read_pickle(behavior_stimulus_file)
rewards = self.get_rewards()
stimulus_presentations = self.get_stimulus_presentations()
rebase_function = self.get_stimulus_rebase_function()
trial_df = get_trials(data, licks, rewards, stimulus_presentations,
rebase_function)
return trial_df
def get_trials(self) -> pd.DataFrame:
"""Get trials from pkl file
Returns
-------
pd.DataFrame
A dataframe containing behavioral trial start/stop times,
and trial data
"""
licks = self.get_licks()
data = self._behavior_stimulus_file()
rewards = self.get_rewards()
stimulus_presentations = self.get_stimulus_presentations()
# Pass a dummy rebase function since we don't have two time streams
trial_df = get_trials(data, licks, rewards, stimulus_presentations,
lambda x: x)
return trial_df