MAINT fix train_evaluator and test_evaluator tests

Author: mfeurer

test/test_evaluation/test_test_evaluator.py

@@ -44,7 +44,7 @@ def test_datasets(self):
                 D_ = copy.deepcopy(D)
                 y = D.data['Y_train']
                 if len(y.shape) == 2 and y.shape[1] == 1:
-                    y = y.flatten()
+                    D_.data['Y_train'] = y.flatten()
                 queue_ = multiprocessing.Queue()
                 evaluator = TestEvaluator(D_, backend_mock, queue_)
 

test/test_evaluation/test_train_evaluator.py

@@ -553,7 +553,7 @@ def test_datasets(self):
                 D_ = copy.deepcopy(D)
                 y = D.data['Y_train']
                 if len(y.shape) == 2 and y.shape[1] == 1:
-                    y = y.flatten()
+                    D_.data['Y_train'] = y.flatten()
                 kfold = ShuffleSplit(n=len(y), n_iter=5, random_state=1)
                 queue_ = multiprocessing.Queue()
                 evaluator = TrainEvaluator(D_, backend_mock, queue_,