From 58a06606766b8e036ee961322ca92566491baa65 Mon Sep 17 00:00:00 2001 From: nicl-nno Date: Mon, 7 Aug 2023 19:29:51 +0300 Subject: [PATCH] test_multi_objective_composer fixed --- test/integration/composer/test_composer.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/test/integration/composer/test_composer.py b/test/integration/composer/test_composer.py index 0d7dc68f27..f5fc5e603b 100644 --- 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']