From 84f4ebb7b9b39f2dfb5712e4ac7df890251597c2 Mon Sep 17 00:00:00 2001 From: Dmitry Gilemkhanov <153055427+dmitryglhf@users.noreply.github.com> Date: Fri, 10 Jan 2025 11:50:40 +0300 Subject: [PATCH] Update evaluation_interfaces.py (#1356) Added check for NoneType in is_multi_output_task --- fedot/core/operations/evaluation/evaluation_interfaces.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) 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