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)