'Auxiliary function to allow compatibility with GAT.'

def _clean_times(self):

    if hasattr(self, 'train_times'):
        self.times = self.train_times
    if hasattr(self, 'train_times_'):
        self.times_ = self.train_times_
    for attr in ['test_times', 'train_times', 'test_times_', 'train_times_']:
        if hasattr(self, attr):
            delattr(self, attr)
    if hasattr(self, 'y_pred_'):
        self.y_pred_ = [y_pred[0] for y_pred in self.y_pred_]
    if hasattr(self, 'scores_'):
        self.scores_ = [score[0] for score in self.scores_]
