'Auxiliary function to allow compatibility with GAT.'

def _prep_times(self):

    self.test_times = 'diagonal'
    if hasattr(self, 'times'):
        self.train_times = self.times
    if hasattr(self, 'times_'):
        self.train_times_ = self.times_
        self.test_times_ = _DecodingTime()
        self.test_times_['slices'] = [[slic] for slic in self.train_times_['slices']]
        self.test_times_['times'] = [[tim] for tim in self.train_times_['times']]
    if hasattr(self, 'scores_'):
        self.scores_ = [[score] for score in self.scores_]
    if hasattr(self, 'y_pred_'):
        self.y_pred_ = [[y_pred] for y_pred in self.y_pred_]
