From 40aabd755b575f9374f214e107845b7463f15883 Mon Sep 17 00:00:00 2001 From: Andrey Stebenkov Date: Wed, 21 Aug 2024 17:57:58 +0300 Subject: [PATCH] Fix bug in tests with IndexError --- .../operation_implementations/implementation_interfaces.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/fedot/core/operations/evaluation/operation_implementations/implementation_interfaces.py b/fedot/core/operations/evaluation/operation_implementations/implementation_interfaces.py index 4a1d2902cc..a08c9a9f12 100644 --- a/fedot/core/operations/evaluation/operation_implementations/implementation_interfaces.py +++ b/fedot/core/operations/evaluation/operation_implementations/implementation_interfaces.py @@ -164,6 +164,8 @@ def _reasonability_check(features): # For every column in table make check if isinstance(features, OptimisedFeatures): features = features._columns + elif isinstance(features, np.ndarray): + features = features.T for column_id, column in enumerate(features): # column = features[:, column_id] if columns_amount > 1 else features.copy()