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