def get_extended_trials(self) -> pd.DataFrame:
"""Get extended trials from pkl file
Returns
-------
pd.DataFrame
A dataframe containing extended behavior trial information.
"""
data = self._behavior_stimulus_file()
return get_extended_trials(data)
def get_extended_trials(self):
filename = self.get_behavior_stimulus_file()
data = pd.read_pickle(filename)
return get_extended_trials(data)