Fix tests

aimclub · Aug 21, 2024 · 5cf8d1b · 5cf8d1b
1 parent 8411c6e
commit 5cf8d1b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git 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
@@ -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)