Fix bug with test_regression_pipeline_with_data_operation_fit_predict…

…_correct by adding check for target
aimclub · Aug 15, 2024 · c123779 · c123779
1 parent ebab7f2
commit c123779
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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