From 35446365fe944aa86f399ec37d9c0603b04e9955 Mon Sep 17 00:00:00 2001 From: Andrey Stebenkov Date: Wed, 21 Aug 2024 17:50:08 +0300 Subject: [PATCH] Fix test_regression_data_operations with inf data after poly_features --- .../operation_implementations/implementation_interfaces.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/fedot/core/operations/evaluation/operation_implementations/implementation_interfaces.py b/fedot/core/operations/evaluation/operation_implementations/implementation_interfaces.py index cb04120612..4a1d2902cc 100644 --- a/fedot/core/operations/evaluation/operation_implementations/implementation_interfaces.py +++ b/fedot/core/operations/evaluation/operation_implementations/implementation_interfaces.py @@ -107,6 +107,8 @@ def transform(self, input_data: InputData) -> OutputData: else: transformed_features = features + transformed_features = np.nan_to_num(transformed_features, copy=False, nan=0, posinf=0, neginf=0) + # Update features and column types output_data = self._convert_to_output(input_data, transformed_features) self._update_column_types(source_features_shape, output_data)