From 5cf8d1bb943587c947b3e0e4684a83b75ecd6480 Mon Sep 17 00:00:00 2001 From: Andrey Stebenkov Date: Wed, 21 Aug 2024 17:14:22 +0300 Subject: [PATCH] Fix tests --- fedot/api/main.py | 1 - .../operation_implementations/implementation_interfaces.py | 7 +++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/fedot/api/main.py b/fedot/api/main.py index 7087b36826..56f19c3b5c 100644 --- a/fedot/api/main.py +++ b/fedot/api/main.py @@ -189,7 +189,6 @@ def fit(self, self.log.message('Final pipeline was fitted') else: self.log.message('Already fitted initial pipeline is used') - # temp # Merge API & pipelines encoders if it is required self.current_pipeline.preprocessor = BasePreprocessor.merge_preprocessors( diff --git a/fedot/core/operations/evaluation/operation_implementations/implementation_interfaces.py b/fedot/core/operations/evaluation/operation_implementations/implementation_interfaces.py index 8822b8e436..cb04120612 100644 --- a/fedot/core/operations/evaluation/operation_implementations/implementation_interfaces.py +++ b/fedot/core/operations/evaluation/operation_implementations/implementation_interfaces.py @@ -5,7 +5,7 @@ import numpy as np from golem.core.log import default_log -from fedot.core.data.data import InputData, OutputData +from fedot.core.data.data import InputData, OutputData, OptimisedFeatures from fedot.core.operations.operation_parameters import OperationParameters from fedot.core.repository.dataset_types import DataTypesEnum from fedot.utilities.custom_errors import AbstractMethodNotImplementError @@ -160,7 +160,10 @@ def _reasonability_check(features): non_bool_ids = [] # For every column in table make check - for column_id, column in enumerate(features._columns): + if isinstance(features, OptimisedFeatures): + features = features._columns + + for column_id, column in enumerate(features): # column = features[:, column_id] if columns_amount > 1 else features.copy() if len(set(column)) > 2: non_bool_ids.append(column_id)