test_multi_objective_composer fixed

aimclub · Aug 7, 2023 · 58a0660 · 58a0660
1 parent e492747
commit 58a0660
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/test/integration/composer/test_composer.py b/test/integration/composer/test_composer.py
@@ -26,7 +26,7 @@
 from fedot.core.repository.operation_types_repository import OperationTypesRepository, get_operations_for_task
 from fedot.core.repository.quality_metrics_repository import ClassificationMetricsEnum, ComplexityMetricsEnum
 from fedot.core.repository.tasks import Task, TaskTypesEnum
-from fedot.core.utils import fedot_project_root
+from fedot.core.utils import fedot_project_root, set_random_seed
 from test.unit.pipelines.test_pipeline_comparison import pipeline_first, pipeline_second
 
 
@@ -162,13 +162,15 @@ def test_parameter_free_composer_build_pipeline_correct(data_fixture, request):
 
 @pytest.mark.parametrize('data_fixture', ['file_data_setup'])
 def test_multi_objective_composer(data_fixture, request):
+    set_random_seed(42)
+
     data = request.getfixturevalue(data_fixture)
     dataset_to_compose = data
     dataset_to_validate = data
     task_type = TaskTypesEnum.classification
     available_model_types = OperationTypesRepository().suitable_operation(task_type=task_type)
     req = PipelineComposerRequirements(primary=available_model_types, secondary=available_model_types,
-                                       max_arity=2, max_depth=2, num_of_generations=1)
+                                       max_arity=2, max_depth=2, num_of_generations=3)
     params = GPAlgorithmParameters(pop_size=2, genetic_scheme_type=GeneticSchemeTypesEnum.steady_state,
                                    selection_types=[SelectionTypesEnum.spea2])
 
@@ -200,7 +202,6 @@ def test_multi_objective_composer(data_fixture, request):
 
 @pytest.mark.parametrize('data_fixture', ['file_data_setup'])
 def test_gp_composer_with_adaptive_depth(data_fixture, request):
-    # TODO: i358 Should be integrational
     data = request.getfixturevalue(data_fixture)
     dataset_to_compose = data
     available_secondary_model_types = ['rf', 'knn', 'logit', 'dt']