diff --git a/fedot/core/operations/evaluation/evaluation_interfaces.py b/fedot/core/operations/evaluation/evaluation_interfaces.py index e0b21e0c1a..23a736cce6 100644 --- 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