Update evaluation_interfaces.py (#1356)

Added check for NoneType in is_multi_output_task
aimclub · Jan 10, 2025 · 84f4ebb · 84f4ebb
1 parent 538f1ba
commit 84f4ebb
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/fedot/core/operations/evaluation/evaluation_interfaces.py b/fedot/core/operations/evaluation/evaluation_interfaces.py
@@ -302,6 +302,7 @@ def convert_to_multivariate_model(sklearn_model, train_data: InputData):
 
 
 def is_multi_output_task(train_data):
-    target_shape = train_data.target.shape
-    is_multi_target = len(target_shape) > 1 and target_shape[1] > 1
-    return is_multi_target
+    if train_data.target is not None:
+        target_shape = train_data.target.shape
+        is_multi_target = len(target_shape) > 1 and target_shape[1] > 1
+        return is_multi_target