Fix bug in tests with IndexError

aimclub · Aug 21, 2024 · 40aabd7 · 40aabd7
1 parent 3544636
commit 40aabd7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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()