From c12377921b957f1238d09f79e48733fe1e48dcc4 Mon Sep 17 00:00:00 2001 From: Andrey Stebenkov Date: Thu, 15 Aug 2024 17:19:04 +0300 Subject: [PATCH] Fix bug with test_regression_pipeline_with_data_operation_fit_predict_correct by adding check for target --- fedot/preprocessing/preprocessing.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/fedot/preprocessing/preprocessing.py b/fedot/preprocessing/preprocessing.py index 64e9720c3d..874c308280 100644 --- a/fedot/preprocessing/preprocessing.py +++ b/fedot/preprocessing/preprocessing.py @@ -603,8 +603,9 @@ def reduce_mem_usage_np(arr, initial_types): self.log.message('-- Reduce memory in features') data.features = reduce_mem_usage_np(data.features, data.supplementary_data.col_type_ids['features']) - self.log.message('-- Reduce memory in target') - data.target = reduce_mem_usage_np(data.target, data.supplementary_data.col_type_ids['target']) + if data.target is not None: + self.log.message('-- Reduce memory in target') + data.target = reduce_mem_usage_np(data.target, data.supplementary_data.col_type_ids['target']) return data