Test fix

test/unit/optimizer/test_data_objective_eval.py

@@ -1,16 +1,14 @@
 import datetime
-
-import pytest
 from copy import deepcopy
 from functools import partial
 
 import numpy as np
+import pytest
 
-from fedot.core.composer.metrics import MSE
 from fedot.core.data.data import InputData
 from fedot.core.data.supplementary_data import SupplementaryData
 from fedot.core.optimisers.fitness import SingleObjFitness
-from fedot.core.optimisers.objective import Objective, PipelineObjectiveEvaluate
+from fedot.core.optimisers.objective import PipelineObjectiveEvaluate
 from fedot.core.optimisers.objective.data_source_splitter import DataSourceSplitter
 from fedot.core.optimisers.objective.metrics_objective import MetricsObjective
 from fedot.core.pipelines.pipeline import Pipeline
@@ -146,7 +144,7 @@ def test_pipeline_objective_evaluate_with_time_constraint(classification_dataset
      throwing_exception_metric]
 )
 def test_pipeline_objective_evaluate_with_invalid_metrics(classification_dataset, metrics):
-    with pytest.raises(ValueError):
+    with pytest.raises(Exception):
         pipeline = sample_pipeline()
 
         data_split = partial(OneFoldInputDataSplit().input_split, input_data=classification_dataset)